BOJ 10799번: 쇠막대기

https://www.acmicpc.net/problem/10799

10799번: 쇠막대기

스택을 이용해서 열린 괄호가 나타나면 스택에 추가하고,

닫힌 괄호가 나타나면 그 전의 괄호가 ( 인 경우에는 스택에서 하나를 pop 한 후에 스택의 길이 즉, 지금까지 있는 쇠막대기의 개수만큼 더해준다. 그 전의 괄호가 ( 가 아닌 경우에는 쇠막대기가 끝나는 것이므로 하나를 더해주고 스택에서 하나를 pop한다.

 

import sys
input = sys.stdin.readline

s = list(input().rstrip())
st = []
ans = 0

for i in range (len(s)):

    if (s[i] == '('):
        st.append(i)
    else:
        if (s[i-1] == '('):
            st.pop()
            ans += len(st)

        else:
            ans += 1
            st.pop()
print(ans)