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

4949번: 균형잡힌 세상

코드

import sys
while 1:
    inp = sys.stdin.readline().rstrip()
    stack = []
    error = 0
    if inp == ".":
        break
    else:
        for i in inp:
            if i == "(":
                stack.append(i)
            elif i == ")":
                if len(stack) == 0:   # 스택이 비어있으면 no 출력
                    error = 1
                    break
                else:
                    if stack[-1] == "(":  # 가장 최근 괄호가 ( 이면 
                        stack.pop()		# 스택에서 pop 해줌
                    else:
                        error = 1
                        break
            elif i == "[":
                stack.append(i)
            elif i == "]":
                if len(stack) == 0:
                    error = 1
                    break
                else:
                    if stack[-1] == "[":
                        stack.pop()
                    else:
                        error = 1
                        break
        if (error == 0) and (len(stack)==0):
            print("yes")            
        else:
            print("no")

 

 

설명

sys.stdin.readlin()으로 문자열을 입력받으면 마지막에 \n이 붙어서 저장되기 때문에 .rstrip()을 추가해서 제거해줘야 된다. 

스택을 이용해서 만약 "("이나 "["이 들어오면 스택에 push를 해주고, ")"나 "]"가 들어오면 현재 스택의 가장 위에 있는 값과 비교하여 같으면 그 값을 pop 해주고 다르면 괄호가 짝이 맞지 않는 것이기 때문에 error 변수를 1로 만들어준 후 break로 for문을 빠져나온다. 

for문을 나온 후 만약 error값이 1이 아니고 스택도 다 비어있으면 yes를 출력해주고 그렇지 않으면 no를 출력해준다.