파이썬 | 백준 | 4949 | 균형잡힌 세상

solution

1. 스택 이용

# 4949, 균형잡힌 세상
import sys

txt = []    # 문자열 입력

while True:
    str = sys.stdin.readline().rstrip()
    if str == '.':
        break
    txt.append(str)

ans = []    # yes, no 저장할 배열

for txt_ in txt:
    check = 1   # 균형 잡힌 문자열이면 1
    stack = []
    for i in txt_:
        if i == '(':
            stack.append('(')
        elif i == ')':
            if len(stack) == 0 or stack[-1] != '(':
                check = 0   # 균형 잡힌 문자열 X
                break
            else:
                stack.pop()
        elif i == '[':
            stack.append('[')
        elif i == ']':
            if len(stack) == 0 or stack[-1] != '[':
                check = 0   # 균형 잡힌 문자열 X
                break
            else:
                stack.pop()

    if len(stack) != 0 or check == 0:   # 스택에 남은 괄호가 있거나 check가 0이면
        ans.append('no')
    else:
        ans.append('yes')


for ans_ in ans:
    print(ans_)

29380KB

108ms

    if len(stack) != 0 or check == 0:
        txt_ = 'no'
    else:
        txt_ = 'yes'

txt_ 변경 안 됨

 

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

4949번: 균형잡힌 세상