파이썬 백준 10799번: 쇠막대기

문제

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

10799번: 쇠막대기

 

풀이

stack을 이용하여 구현하였다.

laser값이 '('인 경우 stack에 append한다.

lasert값이 ')'일 때 두가지 경우가 존재한다.

1. 이전이 '(' 인경우 = 레이저
   이때 stack에 있는 '('의 개수(쇠막대의 개수)만큼 더해준다
2. 이전이 ')'인경우 = 쇠막대기의 끝
   이때는 1만 더한다.(다른 쇠막대에 영향을 주지 않고 자신만 추가)

import sys
laser=sys.stdin.readline()
stack=[]
stick_num=0
last=''
for i in laser:
    if i =='(':
        stack.append(i)
    elif i==')':
        if last==')':
            stick_num+=1
        else:
            stick_num+=(len(stack)-1)
        stack.pop()
    last = i

print(stick_num)