백준: 10799 쇠막대기

문제

10799번: 쇠막대기

 

문제 풀이

 ( ) 연속적으로 괄호가 등장하면 해당 위치는 레이저이다. 따라서 다른 괄호 들과 구분히 되게 ( )를 다른 문자로 치환한다면 문제에 접근하기가 조금 쉬워 진다.

 

 만약 ( )를 ★로 치환하게 된다면 ★(((★★)(★)★))(★) 와 같이 치환되게 된다. 문제는 다음과 같이 풀 수 있다.

1. 여는 괄호인 '('가 입력되면 stack에 `push`하고 닫는 괄호인 ')'가 입력되면 `pop`을 한다.  
2. 현재  값이 여는 괄호 '('라면, 막대기의 수를 1 증가 시킨다.
3. 현재 값이 레이저 라면 stack에 있는 여는 괄호 수만큼 막대기의 수를 증가시킨다.

 즉, stack에 있는 여는 괄호 '('의 수는 중첩된 쇠막대기를 의미한다. 따라서 여는 괄호가 stack에 있고 레이저를 만나게 되면 중첩된 개수만큼 증가시켜주어 문제를 해결 할 수 있다.😊

코드

from sys import stdin


if __name__ == "__main__":
    s = stdin.readline().replace("()", "l")
    stack = []
    answer = 0
    for c in s:
        print(c)
        if c == '(':
            stack.append('(')
            answer += 1
        elif c == ')':
            stack.pop()
        elif c == 'l':
            answer += len(stack)
    print(answer)