[BOJ] 10799번 : 쇠막대기

■ 문제 해설

· 이전 괄호 (prev) 를 이용해서 해결

· '(' 쇠막대기인 경우 stack에 append

· 이전 값 (prev) 와 s[i] 값이 ')'인 경우 => 레이저

   1) 괄호를 맞추기 위해 stack에 pop

   2) stack에 들어간 쇠막대기 수만큼 더함

· 이전 값 (prev) 와 s[i] 값이 ')'인 경우 => 쇠막대기가 끝난 경우

   1) 쇠막대기 절단하여 +1

· 이전 값 (prev) = s[i]로 값 초기화

 

■ 문제 해답

import sys
input = sys.stdin.readline

s = list(input().rstrip())
stack = []
result = 0
prev = None                        

for i in range(len(s)):
  if s[i] == '(':                   
    stack.append(s[i])
  elif prev == '(' and s[i] == ')': 
    stack.pop()                     
    result += len(stack)            

  elif prev == ')' and s[i] == ')': 
    result += 1                     
    stack.pop()
  prev = s[i]

print(result)

문제출처 : https://www.acmicpc.net/problem/10799

10799번: 쇠막대기