[백준] 괄호(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”를 한 줄에 하나씩 차례대로 출력해야 한다. 

 

✨ 풀이 과정

---

💡 이 문제는 스택을 이용해 VPS를 찾는 문제이다.

✅ VPS가 아닌 경우는 두 가지로 정할 수 있다.

1. 괄호가 열리기도 전에 닫는 경우

2. 열린 괄호가 닫히지 않는 경우

 

📖 해결 방법
→ 문자열의 처음부터 반복해가면서 괄호가 열릴 때마다 스택에 push, 닫히면 pop을 하며 짝을 찾으면 된다. 여기서, 스택이 비었는데 pop을 한다거나 문자열이 끝났는데 스택이 비어있지 않다면 VPS가 아니다.

1. 테스트 데이터의 갯수를 입력받는다.

2. 반복문을 통해 테스트 데이터를 입력받는다.

→ sys 모듈을 쓸 경우 개행문자도 문자열에 포함되기 때문에 rstrip()을 통해 제거해줌.

3. "("을 만났을 경우 stack에 추가한다.

4. "("이 아닌 경우 else 구문을 실행한다.

5. 오류가 발생할 가능성이 있는 코드 → 즉 ")"을 만났을 때 pop()을 통해 제거해준다.

6. 에러가 발생했을 경우 시행할 코드 → 즉, stack에 아무것도 없어서 pop()을 실행하지 못할 경우에 No를 출력해준다.

7. 반복문을 통해 모든 테스트 데이터를 다 돌았을 때 스택이 비어있지 않은 경우에는 NO를 출력하고 아닌 경우 Yes를 출력한다.

 

📖 다른 풀이 방법

알고리즘 스터디를 하다가 얻은 아이디어인데, while s in 문자열: 의 반복을 통해 () 쌍이 모두 제거될 때까지 replace() 함수를 돌리는 방법도 있다고 하였다.

· 참고할 사이트 : https://ooyoung.tistory.com/77

for _ in range(int(input())):
    b = input()
    while '()' in b:
        b = b.replace('()','')
    print("NO" if b else "YES")

 

👩‍💻 최종 코드

---

import sys
input = sys.stdin.readline

#테스트 데이터의 갯수를 입력받는다.
T=int(input())

for i in range(T):
    stack = []
    #반복문을 통해 테스트 데이터를 입력받는다.
    #sys 모듈을 쓸 경우 개행문자도 문자열에 포함되기 때문에 rstrip()을 통해 제거해준다.
    for i in input().rstrip():
        #"("을 만났을 경우 stack에 추가한다.
        if i == "(":
            stack.append(i)
        #"("이 아닌 경우 else구문을 실행한다.
        else:
	    #오류가 발생할 가능성이 있는 코드->즉, ")"을 만났을 때 pop()을 통해 제거해준다.
            try:
                stack.pop()
	    #에러가 발생했을 경우 시행할 코드->즉, stack에 아무것도 없어서 pop()을 실행하지 못할 경우에 No를 출력해준다.
            except:
                print("NO")
                break
    #반복문을 통해 모든 테스트 데이터를 다 돌았을 때 시행한다.
    else:
        # 스택이 안 비어있는 경우에는 NO를 출력하고 아닌 경우 Yes를 출력한다.
        print("NO" if stack else "YES")