[백준 - python] 4949번: 균형잡힌 세상

문제 링크

4949번: 균형잡힌 세상

문제 풀이

while 1:
    stack = []
    str = input()

    if str == ".":
        break

    for letter in str:
        if letter == "[" or letter == "(":
            stack.append(letter)

        if letter == "]":
            if len(stack) != 0 and stack[-1] == "[":
                stack.pop()
            else:
                stack.append(letter)

        if letter == ")":
            if len(stack) != 0 and stack[-1] == "(":
                stack.pop()
            else:
                stack.append(letter)

    if len(stack) == 0:
        print("yes")
    else:
        print("no")

4949 입출력 예시

*key point: 9012번 문제와 유사하나 괄호의 종류가 늘어났다. 스택과 조건문을 활용해서 정상적으로 같은 종류의 괄호가 닫힐 때를 찾는다.