티스토리 뷰

PS/Python

[Python] 백준 #4949 균형잡힌 세상 - stack

yoo.o 2020. 8. 13. 15:18
반응형

문제

 

4949번: 균형잡힌 세상

문제 세계는 균형이 잘 잡혀있어야 한다. 양과 음, 빛과 어둠 그리고 왼쪽 괄호와 오른쪽 괄호처럼 말이다. 정민이의 임무는 어떤 문자열이 주어졌을 때, 괄호들의 균형이 잘 맞춰져 있는지 판단

www.acmicpc.net

풀이

괄호 문제라 스택을 이용해서 풀었다.

stack = []

while True:
    stack.clear()
    flag = 0
    line = input()
    if line == ".":
        break

    for each in line:
        if flag == 1:
            break

        if each == "(" or each == "[":
            stack.append(each)
        elif each == ")":
            if len(stack) > 0 and stack[-1] == "(":
                stack.pop()
            else:
                flag = 1
        elif each == "]":
            if len(stack) > 0 and stack[-1] == "[":
                stack.pop()
            else:
                flag = 1

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

yes/no로 출력해야하는데 YES/NO로 짜서 자꾸 틀렸다고 나와서 삽질함 ..

 

반응형
저작자표시