[파이썬] 백준 9012번 : 괄호

출처 : www.acmicpc.net/problem/9012

9012번: 괄호

stack을 이용하여 '(' 이 나오면 stack에 넣고 ')'가 나오면 stack에서 빼는 방법을 생각했다.

만약, stack에 아무 것도 없는데 ')'이 나오면 무조건 'NO'이므로 그런 경우에는 따로 처리해주었다.

import sys
 
N = int(sys.stdin.readline().strip())
 
for _ in range(N):
    s = sys.stdin.readline().strip('\n')
    j = 0
    stack = []
    idx = 0
    while(1):
        if idx >= len(s):
            if len(stack) != 0:
                j = 1
            break
        if s[idx] == '(':
            stack.append(idx)
            idx += 1
        else:
            if len(stack) == 0:
                j = 1
                break
            stack.pop()
            idx += 1
    if j == 1:
        print('NO')
    else:
        print('YES')