코딩 테스트 (Python)/백준

[백준] 9012번 괄호 (Python)

hihyuk 2024. 1. 4. 13:57
 

9012번: 괄호

괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고

www.acmicpc.net

더보기

문제

괄호 문자열(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. 입력받을 테스트 케이스의 개수를 입력받는다.
  2. 괄호를 입력받을 때마다 결과 값을 출력한다.
  3. 입력 받은 괄호를 맨 앞부터 확인 후 '(' 라면 리스트에 저장한다.
  4. ')' 라면 리스트에 '(' 있을 경우 제거해주고 그렇지 않다면 NO를 출력 한다.
  5. 모두 확인 후 리스트가 비었다면 YES를 출력하고 그렇지 않다면 NO를 출력한다.

 

예제 입력

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

예제 출력

NO
NO
YES
NO
YES
NO

 

풀이

T = int(input())

for _ in range(T):
    line = input()
    stack = []
    for i in line:
        if i == '(':
            stack.append(i)
        elif i == ')':
            if stack:
                stack.pop()
            else:
                print("NO")
                break
    else:
        if not stack:
            print("YES")
        else:
            print("NO")

 

한 줄씩 해석해보기

T = int(input())
  • 테스트 데이터 개수 입력 받기
for _ in range(T):
    line = input()
    stack = []
  • 데이터 개수 만큼 괄호를 입력받을 때 마다 stack생성
for i in line:
    if i == '(':
        stack.append(i)
  • 입력받은 괄호 하나씩 검사
  • '(' 일 경우 stack에 추가
elif i == ')':
    if stack:
        stack.pop()
    else:
        print("NO")
        break
  • ')'일 경우 stack에 값이 있으면 제거
  • stack에 값이 없을시 NO 출력 후 break
else:
    if not stack:
        print("YES")
    else:
        print("NO")
  • break문으로 끊기지 않고 수행됐을경우 수행
  • break문으로 안끊기고 스택이 비어있다면 괄호가 다 맞기때문에 YES 출력
  • break문에 걸리지 않았더라도 스택에 괄호가 들어있다면 NO 출력