DataStucture1_02괄호(9012)

문제

괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고 부른다. 한 쌍의 괄호 기호로 된 “( )” 문자열은 기본 VPS 이라고 부른다. 만일 x 가 VPS 라면 이것을 하나의 괄호에 넣은 새로운 문자열 “(x)”도 VPS 가 된다. 그리고 두 VPS x 와 y를 접합(concatenation)시킨 새로운 문자열 xy도 VPS 가 된다. 예를 들어 “(())()”와 “((()))” 는 VPS 이지만 “(()(”, “(())()))” , 그리고 “(()” 는 모두 VPS 가 아닌 문자열이다.

여러분은 입력으로 주어진 괄호 문자열이 VPS 인지 아닌지를 판단해서 그 결과를 YES 와 NO 로 나타내어야 한다.

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 주어진다. 입력의 첫 번째 줄에는 입력 데이터의 수를 나타내는 정수 T가 주어진다. 각 테스트 데이터의 첫째 줄에는 괄호 문자열이 한 줄에 주어진다. 하나의 괄호 문자열의 길이는 2 이상 50 이하이다.

출력

출력은 표준 출력을 사용한다. 만일 입력 괄호 문자열이 올바른 괄호 문자열(VPS)이면 “YES”, 아니면 “NO”를 한 줄에 하나씩 차례대로 출력해야 한다.

풀이

• stack 이용해서 여는 괄호 나오면 배열에 "(" 추가
• 닫는 괄호 나오면 pop 하기
• 만약 배열의 len 이 0인데도 pop 을 요구하면 결과값은 NO
• for문이 다 돌고 괄호를 모두 확인한 후 len 값이 0이면 YES 아니면 NO

코드

import sys
sys.stdin = open("input.txt", "rt")


def input():
    return sys.stdin.readline().rstrip()

N = int(input())
res = 0

for i in range(N):
    PS = input()
    arr=[]

    val = 0
    for x in PS:        
        if x == "(":
            arr.append("(")
        else :
            if len(arr)==0:
                val=-1
                break
            arr.pop()

    if val == 0:
        if len(arr) == 0:
            print("YES")    
        else:
            print("NO")
    else: print("NO")

배운 것

코멘트

예전에 풀었을 때 배열에 append 시킬때 "("말고 1로 해서 len을 확인할 필요 없이 0보다 작으면 NO를 출력하도록 하고, for문 다 돌았을 때 상태가 0이면 YES 아니면 NO를 출력하도록 했었는데 그게 더 나은듯...
그리고 앞으로는 미리미리좀 하자 밤 늦게 뭐하니 ^^?