Algorithm/Baekjoon

[Python3] 백준(Baekjoon) 괄호 9012

앵도라지 2022. 3. 3. 23:32

문제 :

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

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

접근 :

'('가 입력되면 push, ')'이 입력되면 pop을 한다.

이 때, 스택이 비어있는데 pop을 해야 하는 경우를 대비해, 2가지의 조건을 구분한다.

스택에 아무것도 없는데 ')'가 입력된 경우에는 'X'라는 임의의 문자열을 삽입한다.

풀이 :

T = int(input())
for tc in range(1, T+1):
    arr = list(input())
    ans = []
    for i in range(len(arr)):
        if arr[i] == '(':
            ans.append(arr[i])
        else:
            if len(ans) > 0 and ans[-1] == '(':
                ans.pop()
            else:
                ans.append('X')
                break

    if len(ans) != 0 :
        print('NO')
    else:
        print('YES')