백준 (boj) 4949 파이썬 - 균형잡힌 세상

문제 링크 : https://www.acmicpc.net/problem/4949

4949번: 균형잡힌 세상

 

 

스택으로 풀면 되는 문제였다. (가장 나중에 넣은 데이터를, 가장 먼저 삭제하는 방식)

"( [ [ [ ( " 이러한 괄호가 들어왔다면 , 제일 마지막에 들어온 괄호순서대로 소거하는 방식으로 코딩을 하면 된다.

import sys
input = sys.stdin.readline

while(True):
    arr =  []
    string = input().rstrip()
    if string == '.' : break
    else: 
        for i in string :
            #여는 괄호를 arr에 추가
            if i == '[' : arr.append('[')
            elif i == '(' : arr.append('(')
            
            #닫는괄호가 arr의 마지막에 들어온것과 일치하는지 확인
            elif i == ']' :
                if len(arr)==0 : arr.append("no!");break
                elif arr[-1] == '[' : arr.pop()
                else : break
            elif i == ')' : 
                if len(arr)==0 : arr.append("no!");break
                elif arr[-1] == '(' : arr.pop()
                else :  break
    #출력부분
    if len(arr)==0: print('yes')
    else : print('no')