[백준] 9012 괄호 -python

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

9012번: 괄호

내가 처음에 짠 코드

n = int(input())

for i in range(n):
    x = list(input())
    l_cnt = 0
    r_cnt = 0
    for i in range(len(x)):
        if x[i] == '(':
            l_cnt += 1
        else : 
            r_cnt += 1
    if l_cnt == r_cnt :
        if x[-1] == ')':
            print('YES')
    else :
        print('NO')

그냥 단순하게 ( ) 의 개수가 같은지, 끝이 )로 끝나는지만 체크함.

테스트케이스 1은 잘 돌아갔는데, 2에서  ())(()에 걸려버렸다.

 

어떻게하지 고민했는데 stack을 이용하면 쉬운거였음

(가 들어올때마다 stack에 쌓고, )가 들어오면 pop으로 빼기

 

for - else문을 활용하여 두가지 조건을 걸고 풀었다.

1. 반복문이 다 돌았는데, stack이 비어있으면 YES / 아니면 NO

2. stack 안에 들어있는 (가 없는데 ) 가 들어와서 pop해야 하면 NO

 

n = int(input())

for i in range(n):
    x = list(input())
    stack = []
    for i in range(len(x)):
        if x[i] == '(':
            stack.append(x[i])
        else :
            if len(stack) != 0:
                stack.pop()
            else :
                print('NO')
                break
    else : 
        if len(stack) == 0:
            print('YES')
        else :
            print('NO')