백준 10799 - 쇠막대기 [파이썬]

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

10799번: 쇠막대기

 

풀이 과정

count = 아직 안끝난 쇠막대기의 개수

answer = 정답 이라고 하자.

 

'(' 인식시 일단 쇠막대기의 시작이라고 간주 => count += 1

 

')' 인식시 앞 문자가 '('이면 앞의 '('이 쇠막대기의 시작이 아니였으므로 count -= 1, 그 후 안끝난 쇠막대기들 자르니까 answer += count를 해주고, 앞 문자가 '('이 아니면 쇠막대기의 끝이므로 count -= 1, answer += 1를 해주면 된다.

 

코드

import sys

solve = list(input())
count = 0
answer = 0
for i in solve:
    if i == '(':
        count += 1
    elif i == ')':
        if stack[-1] == '(':
            count -= 1
            answer += count
        else:
            count -= 1
            answer += 1

print(answer)