[백준] 9012번 - 괄호 (Python)

https://www.acmicpc.net/problem/9012

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

 

 

<문제 풀이>

1. 시작 괄호 '('를 만나면 일단 스택에 삽입한다.

2. 그렇게 진행을 하다가 ')'를 만나면 연산을 시작한다.

3. vps가 되는 조건은 '(' ')' 괄호가 짝이 맞아야 한다.

4. ')'를 만나면 스택에서 '('을 하나 꺼내서 vps를 만들어서 날린다.

5. 마지막까지 진행했을 때 스택에 '('가 남아 있거나, 스택에는 없는데 마지막이 ')'이면 vps가 아니므로 'NO'를 출력한다.

 

<코드>

for _ in range(int(input())):
    # vps의 조건은 마지막까지 괄호 짝이 맞아야 한다. 
    stk = []
    ans = 'YES'
    for c in input():
        if c == '(': # '('을 만나면 일단 스택에 넣는다.
            stk.append(c)
        else: # ')'을 만나면 
            if len(stk) > 0: # 스택에 시작 괄호가 남아 있으면
                stk.pop() # 쌍을 맞춰서 제거한다
            else:
                ans = 'NO' # 스택안에 시작 괄호가 없으면 짝이 안 맞으므로 vps가 아니다
        if len(stk) > 0:
            ans = 'NO' # 연산을 다 마쳤는데 스택안에 시작 괄호가 남아 있으면 vps가 아니다.

        print(ans)