[백준] 10799번 쇠막대기

문제링크

스택 활용하기

문자열 입력받기

반복문을 통해 입력 받은

w[i]가 ‘(‘ 라면 스택에 추가

else

w[i가] ‘)’ 라면

w[i-1]이 ‘(‘ 이면 레이저 -> 스택의수 (len(s)) 만큼 카운트 추가

w[i-1]이 ‘)’ 이면 막대기의 끝 -> 막대기의 끝 카운트 +1

파이썬

w = list(input())
s = []
cnt =0
for i in range(len(w)):
    if w[i] == '(':
        s.append('(')
    else:
        if w[i-1] == '(':
            s.pop()
            cnt+= len(s)
        else:
            s.pop()
            cnt+=1
print(cnt)