백준 4949 (python): 균형잡힌 세상

import sys

while True:
    cmd = list(sys.stdin.readline().rstrip())
    if cmd[0] == '.':
        break
    stack = []
    for i in cmd:
        if i == '(' or i == '[':
            stack.append(i)
        elif i == ')':
            if not stack or stack.pop() != '(':
                print("no")
                break
        elif i == ']':
            if not stack or stack.pop() != '[':
                print("no")
                break
        elif i == '.':
            if stack:
                print("no")
            else:
                print("yes")

 

우와! 이것도 전에 비해 3배나 성능 향상!~~~~~~~~ 

마지막 테스트 케이스 ' .' → 공백 제거 방지 위해 rstrip()

 

* rstrip() 

: 오른쪽 공백을 삭제

 

* lstrip()

: 왼쪽 공백을 삭제

 

* strip()

: 왼쪽, 오른쪽 공백을 삭제