[파이썬] 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

 

 

😘 나의 풀이

 

for i in range(int(input())):
    PS = input()
    stack = list()
    for p in PS:
        if p == '(':
            stack.append('(')
        else:
            if stack:
                stack.pop()
            else:
                print("NO")
                break
    else:       # break가 실행된 적 없으면
        if stack:       # 값이 남아있으면
            print("NO")
        else:
            print("YES")

 

스택을 이해하고 있으면 쉽게 풀 수 있다.

'('면 stack에 넣었다.

')'면 stack에서 pop한다.

stack에서 괄호를 pop해야하는데 비어있으면 NO를 출력하고 break한다.

break가 실행된 적 없다면 stack에 값이 남아있는지 본다.

짝이 맞다면 반복문이 끝난 후 stack이 비어있어야 한다.

 

 

🙂 다른 사람 풀이

 

from sys import stdin

n=int(input())
for _ in range(n) :
    str_=stdin.readline().strip()
    stack=0
    for chr_ in str_ :
        if chr_ == '(' :
            stack+=1
        else :
            stack-=1
            if stack < 0 :
                break
    if stack == 0 :
        print('YES')
    else :
        print('NO')

 

stack에 직접 넣고 빼지 않고 갯수를 파악해서 풀었다.

좀 더 효율적이다.

 

 

 

문제 출처 💁‍♀️ 백준