[백준] 10799번:쇠막대기 (python 파이썬)

www.acmicpc.net/problem/10799

10799번: 쇠막대기

알고리즘 및 소스코드

1. '(' 가 나오면 스택에 추가

2. '()' 가 나오면 결과 값에 스택 길이 더하기

3. ')' 가 나오면 결과 값에 1 더하고 pop 한 번 해주기

arr = input()

stack = []
result = 0
for i in range(len(arr)):
    if arr[i] == '(':
        stack.append(arr[i])
    else:
        if arr[i-1] == '(':
            stack.pop()
            result += len(stack)
        else:
            stack.pop()
            result += 1
print(result)