백준 9012번 - 괄호 - 파이썬(Python)

문제 링크

https://www.acmicpc.net/problem/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”를 한 줄에 하나씩 차례대로 출력해야 한다. 

 

입력 예시1

6
(())())
(((()())()
(()())((()))
((()()(()))(((())))()
()()()()(()()())()
(()((())()(

출력 예시1

NO
NO
YES
NO
YES
NO

입력 예시2

3
((
))
())(()

출력예시2

NO
NO
NO

풀이 전 생각

VPS 가 되는 조건에 대해서 생각했다.

 

1. ')' 가 오기 위해서는 앞에 '('가 무조건 존재해야한다.

2. 그리고 앞에 있는 '('의 개수가 ')'의 개수보다 더 많거나 같아야 한다.

 

위 두가지 조건을 생각해봤을 때,

1. 스택을 사용해서 '('가 오면 스택에 넣고,

2. ')'가 왔을 때는 '('를 pop 해주는 방식으로 구현하면 될 것 같다. 주의점은 ')'가 왔을 때 stack이 비어있다면 앞에 '('가 없다는 뜻이므로 조건을 만족하지 못한다.  

 

예제 코드

n = int(input())

for i in range(n):
    s = input() 
    stack = []
    is_valid = True  # VPS 인지 아닌지 체크하기 위함
    
    for char in s:
        if char == '(':
            stack.append(char)
        else:
            if stack:
                stack.pop()
            else:
                is_valid = False  # ')'인데 앞에 '('가 없는 경우
                break
    
    if is_valid and not stack:
        print('YES')
    else:
        print('NO')

 

 

생각