목차
문제
괄호 문자열(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”를 한 줄에 하나씩 차례대로 출력해야 한다.
예제
입력
6
(())())
(((()())()
(()())((()))
((()()(()))(((())))()
()()()()(()()())()
(()((())()(
Plain Text
복사
출력
NO
NO
YES
NO
YES
NO
Plain Text
복사
풀이
import sys
# vps 판단 function
def ps_stack(ps_list):
stack_count = 0
for p in ps_list:
if p == '(':
stack_count += 1 # '('가 하나 들어오면 ')'가 하나 들어올 수 있음
else:
if stack_count == 0: # ')'가 들어왔는데 만약 '('가 하나도 없다면 VPS가 될 수 없음.
return 'NO'
else: # 들어올 수 있는 ')'가 하나 줄어듦
stack_count -= 1
if stack_count == 0: # '('와 ')' 개수가 동일하면 VPS
return 'YES'
else:
return 'NO'
# input
M = int(sys.stdin.readline())
is_vps = []
for i in range(M):
ps_list = list(input())
is_vps.append(ps_stack(ps_list))
for i in is_vps:
print(i)
Python
복사
기억할 점
•
list(input())을 사용하면 연속으로 들어오는 문자열을 원소 하나씩 쪼개서 list로 저장할 수 있다.
•
그런데 이 때 list(sys.stdin.realine())을 하면 안 먹음