백준 알고리즘

백준 9012번 괄호 문제풀이, 파이썬

고인호 2023. 5. 9. 15:55
반응형

백준 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

 

 

<작성한 코드>

 

T = int(input())
for _ in range(T):
  VPS = input()
  cnt = 0
  for i in VPS:
    if i == '(':
      cnt +=1
    elif i ==')':
      cnt -=1
      if cnt < 0:
        break
  if cnt == 0:
    print('YES')
  else:
    print('NO')

 

 

<코드 풀이>

문제를 처음 읽었을 때는 이걸 어떻게 구현하지... 하고 어렵게 느껴졌었는데

조금만 생각해보니 보기보다 쉽게 풀리는 문제였다.

 

 

T = int(input())
for _ in range(T):
  VPS = input()
  cnt = 0

우선 위와 같이 초기 세팅을 해주었다. 

VPS라는 이름의 변수에 괄호 문자열을 입력받고

cnt에는 초기값 0을 넣어주었다. 

 

문제에서 한 쌍의 괄호 기호로 된 "( )"이 기본 문자열 이기 때문에 

'(' 가 나오면 cnt에 +1을, ')' 가 나오면 cnt에 -1을 하는 방향으로 생각했다. 

 

 

 

  for i in VPS:
    if i == '(':
      cnt +=1
    elif i ==')':
      cnt -=1
      if cnt < 0:
        break
  if cnt == 0:
    print('YES')
  else:
    print('NO')

위처럼 입력받은 괄호 문자열을 처음부터 하나씩 반복문을 돌면서

나오는 괄호 모양에 따라 +1 또는 -1을 해준다. 

그런데 이때 cnt가 음수가 되는 순간 반복문을 멈추도록 했다. 

괄호 문자열이 한쌍이 되려면 기본적으로 ) 전에 (가 있어야하므로 

음수가 나오는 순간 한쌍이 성립할 수 없게 되기 때문이다. 

반응형