[백준 4949번 ] 균형잡힌 세상 (파이썬/python)

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

4949번: 균형잡힌 세상

 

스택을 이용하면 쉽게 풀 수 있는문제.

마지막에 스택에 남아있는 괄호가 있는지까지 판단해야지 된다!

 

만약 이걸 판단 안 할 경우

입력값으로 [[ 같은것들이 들어와도 yes를 반환하니까!

 

 

while 1:
    input_str = input()
 
    if input_str == '.':
        break
    st = []
    result = True
    for s in input_str:
        if s == '[' or s == '(':
            st.append(s)
        elif s == ']':
            if st and st[-1] == '[':
                st.pop()
            else:
                result = False
                break
        elif s == ')':
            if st and st[-1] == '(':
                st.pop()
            else:
                result = False
                break
    if not result or st:
        print('no')
    elif result:
        print('yes')