[백준] 9012번 괄호 (Python)

문제

www.acmicpc.net/problem/9012

9012번: 괄호

 

코드

T = int(input())

for i in range(T):
    stack = []
    a=input()
    for j in a:
        if j == '(':
            stack.append(j)
        elif j == ')':
            if stack:
                stack.pop()
            else: # 스택에 괄호가 없을경우 NO
                print("NO")
                break
    else: # break문으로 끊기지 않고 수행됬을경우 수행한다
        if not stack: # break문으로 안끊기고 스택이 비어있다면 괄호가 다 맞는거다.
            print("YES")
        else: # break안 걸렸더라도 스택에 괄호가 들어있다면 NO이다.
            print("NO")

 

코드설명

'(' 괄호가 들어오면 stack에 넣어준다.

')' 괄호가 들어왔을때는 stack을 검사해 비어있다면 NO를 출력하고 break해주고 만약 비어있지 않다면 pop을 한번 해준다.

 

for-else 문으로 for문에서 한번도 break가 난적이 없다면 else문을 실행하는데 그럴경우 stack이 비어있다면 괄호가 모두 짝이 맞은 경우이기 때문에 YES를 출력한다.

else 문이 실행해도 스택안에 괄호가 남아있다면 짝이 안맞는 것이기 때문에 NO를 출력한다.