Agorithm/백준 알고리즘
백준 알고리즘 10799번 쇠막대기(python)
kimjinho1
2021. 3. 7. 00:12
728x90
반응형
기본적인 스택 문제이다. 딱 작년만 해도 이 문제를 못 풀었었는데 이제는 쉽게 풀 수 있게 되었다. 굿!!
from collections import deque
s = input()
stack = deque()
res = 0
for c in s:
if c == '(':
stack.append(c)
else:
if stack[-1] == '(':
stack.pop()
stack.append(1)
else:
t = 0
while stack[-1] != '(':
t += stack.pop()
stack.pop()
stack.append(t)
res += t+1
print(res)
728x90
반응형