문제
괄호 문자열(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”를 한 줄에 하나씩 차례대로 출력해야 한다.
내 풀이
input1 = int(input())
def VPS(str1):
stack = []
for s in str1:
if s == '(':
stack.append(s)
else: # s ==')'
if len(stack)==0:
print('NO')
break
else:
stack.pop()
else: # break 문으로 중단되지 않았을 경우 수행
if len(stack)==0:
print('YES')
else:
print('NO')
for i in range(input1):
VPS(input())코멘트
주요 아이디어는 닫힌 괄호(')')가 임의 위치의 열린 괄호('(')와 상쇄되는 것이 아니라 닫힌 괄호가 나오기 바로 전의, 즉, 가장 가까운(최근) 열린 괄호와 상쇄된다는 것이다. 가장 가까운(최근)이라는 키워드를 보고 스택을 떠올릴 수 있어야 한다.
문자열의 가장 처음부터 살펴보면서 열린 괄호로 시작하면 stack에 push하고, 닫힌 괄호로 시작하면 볼 것도 없이 'NO'를 출력하도록. 순차적으로 보면서 열린 괄호 다음에 닫힌 괄호가 오게 되면 짝이 맞춰졌으므로 stack을 pop해서 empty stack으로 만들어준다.
그렇게 마지막에 stack이 비었으면 짝이 다 맞는 것이므로 'YES' 출력, 그렇지 않다면 짝이 맞지 않으므로 'NO'. 반드시 for-else를 사용해서 짝이 맞지 않아서 empty stack이 되는 경우와 짝이 딱 맞아서 empty stack이 되는 경우를 구분해 주도록 한다.
References
https://www.acmicpc.net/problem/9012
https://wikidocs.net/223100
https://wookcode.tistory.com/49
