[백준 4949] 균형잡힌 세상 - 파이썬

문제

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

4949번: 균형잡힌 세상

 

 

 

풀이

while(True):
    sentence = input()
    if(sentence == "."):
        break
    stack = []


    for i in sentence:
        if(i == "(" or i == "["):
            stack.append(i)
        if(i == ")"):
            if(len(stack) != 0 and stack[-1] == "("):
                stack.pop()
            else:
                stack.append(")")
                break
        if(i == "]"):
            if(len(stack) != 0 and stack[-1] == "["):
                stack.pop()
            else:
                stack.append("]")
                break


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

 

1. 입력받은 문장에서 괄호만 찾는다.

2. 만약 괄호가 여는 괄호면 스택에 추가한다. 

3. 만약 괄호가 닫는 괄호면 경우의 수가 두 개 있는데, 

3-1. 여는 괄호가 추가된 상태에서 닫는 괄호가 들어온 경우 여는 괄호를 스택에서 제거한다. 

3-2. 여는 괄호가 없는 상태에서 닫는 괄호가 들어온 경우 닫는 괄호를 스택에 추가하고 for문을 종료한다.

4. 만약 스택이 비어있으면 yes를, 스택이 비어있지 않으면 no를 출력한다.