Algorithm/백준

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

potato_pizza 2024. 6. 21. 15:23
728x90

균형잡힌 세상(4949)

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

 

문제

코드

  • 스택을 활용한 문제
  • ']'나 ')'가 나오면 stack내에 '[' , '('에 따라서 pop을 실시
  • stack내 남은게 없다면 yes, 남은게 있으면 no
# 균형잡힌 세상
# 4949

while True :
    word = input()
    stack = []

    if word == "." :
        break

    for i in word :
        if i == '[' or i == '(' :
            stack.append(i)
        elif i == ']' :
            if len(stack) != 0 and stack[-1] == '[' :
                stack.pop() 
            else : 
                stack.append(']')
                break
        elif i == ')' :
            if len(stack) != 0 and stack[-1] == '(' :
                stack.pop()
            else :
                stack.append(')')
                break
    if not stack:
        print('yes')
    else :
        print('no')
728x90
반응형