[백준]9012번 괄호 - 스택

 

n = int(input())
def vps(p):
    stack = []
    for i in p:
        if i == "(":
            stack.append(i)
        else:
            if not stack:
                print("NO")
                return
            else:
                stack.pop()
    if not stack:
        print("YES")
        return
    else:
        print("NO")
        return
    
for i in range(n):
    p = input()
    vps(p)

 

코드 설명

반복 횟수를 n 변수에 입력 받는다. 그 후 반복문을 통해 괄호를 입력받고, 함수를 실행한다.

만약 문자가 "(" 라면 스택에 넣는다. 문자가 ")" 일때 만약 스택에 아무 것도 없다면 "NO"를 출력한다. 스택에 값이 존재한다면 "("를 pop 한다.

반복문을 돌고 난 후 스택이 비어있다면 "YES" 비어있지 않다면 "NO"를 출력한다.

 

Point: 스택에는 "(" 만 저장하는 것이 포인트이다.

주의 : 스택이 비어있을 경우 ")" 가 들어올 경우는 VPS 성립 X, 반복문을 수행 후 스택이 비어있다면 VPS 성립 O, 비어있지 않다면 성립 X