BOJ : 10799. 쇠막대기 (python)

BOJ 10799 문제 링크

10799번: 쇠막대기

풀이

()는 레이저이므로 ()를 다른 문자로 replace 하여 바꿔주었다.

만약 * 이면 레이저이므로 스택에 들어있는 막대기가 한번 잘리므로 res에 len(stack)을 더해 줌.

stack에는 (밖에 없으므로 )가 들어왔을 때에는 하나의 막대기로 생각하여 pop해준다.

코드

#10799. 쇠막대기
iron = input().replace('()','*')
stack, res = [], 0
for i in iron:
    if i == '(':
        stack.append(i)
    elif i == '*':
        if stack:
            res += len(stack)
    elif i == ')':
        stack.pop()
        res += 1
print(res)