[백준] 4949 균형잡힌 세상 (Python)

[백준] 4949 균형잡힌 세상 (Python)

 

Source

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

4949번: 균형잡힌 세상

 

Code

import sys

input = sys.stdin.readline

while True:
    line = str(input().rstrip())
    stack = []

    for i in line:
        if i == '(':
            stack.append('(')
        elif i == '[':
            stack.append('[')
        elif i == ')':
            if len(stack) > 0 and stack[-1] == '(':
                stack.pop()
            else:
                stack.append(')')
        elif i == ']':
            if len(stack) > 0 and stack[-1] == '[':
                stack.pop()
            else:
                stack.append(']')

    if line == '.':
        break

    if stack:
        print('no')
    else:
        print('yes')

 

How to solve?

()나 []처럼 괄호가 열린 뒤 닫혀야 한 쌍이 성립한다.

따라서 스택 구조를 만들어서 push가 이루어진 뒤 만약 )나 ]를 push했는데 그 이전에 (나 [가 있어 쌍구조를 이룬다면 pop해준다. 이렇게 만든 스택이 비었다면 yes를 출력해주고 비어있지 않다면 no를 출력해준다.