백준 4949 - 균형잡힌 세상

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

4949번: 균형잡힌 세상

ans = []
while True:
    text = input()
    if text=='.':
        break
    stack = []
    flag = "yes"
    for word in text:
        if word=='(' or word=='[':
            stack.append(word)
        elif word==')':
            if not stack:
                flag="no"
                break
            if stack[-1]=='(':
                stack.pop(-1)
            else:
                flag="no"
                break
        elif word==']':
            if not stack:
                flag="no"
                break
            elif stack[-1]=='[':
                stack.pop(-1)
            else:
                flag="no"
                break
        elif word=='.':
            break
    if stack:
        flag="no"
    ans.append(flag)

for word in ans:
    print(word)

이전의 괄호 문제의 업그레이드 버전이다.

괄호가 하나 추가된 것 말고는 별 차이는 없다.

여전히 체크해야 하는건

-')' 나 ']'가 입력되었을 때

1. stack에 값이 있는지 체크

2. stack[-1]이 적절한 괄호인지 체크

- 반복문이 끝났을 때

1. stack에 값이 있는지 체크

 

하는 것이 중요하다.