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

 

입출력 예제

입력	출력
6 (())()) (((()())() (()())((())) ((()()(()))(((())))() ()()()()(()()())() (()((())()(	NO NO YES NO YES NO
3 (( )) ())(()	NO NO NO

 

풀이

VPS의 조건은 다음과 같다.

1. 여는 괄호와 닫는 괄호의 수가 동일함
2. 맨 앞에서부터 괄호의 수를 셀 때 여는 괄호의 수가 항상 닫는 괄호의 수보다 많거나 같아야 함

스택을 이용하여 VPS를 검사하는 로직은 다음과 같다.

1. 여는 괄호를 만나면 스택에 넣음
2. 닫는 괄호를 만나면 스택이 비어있지 않은 경우 스택에서 여는 괄호를 하나 삭제하고(pop), 스택이 비어있는 경우 ‘NO’를 출력
3. 주어진 문자열의 맨 마지막 문자까지 검색했을 때 스택이 비어있는 경우 ‘YES’를 출력, 그렇지 않은 경우 ‘NO’를 출력-- 여는 괄호와 닫는 괄호의 수가 동일하다면 맨 마지막에 스택이 비어있어야 함
4. 주어진 테스트케이스에 대해 1~3을 반복

시행착오

• 스택과 valid 변수는 각 테스트케이스마다 초기화될 수 있도록 for문 안쪽에 선언해야 함
• 2에서 NO를 출력하는 것과 3에서 맨 마지막 스택 검사 후 Y/N 출력하는 것이 겹쳐서 두 번 결과가 출력되는 문제가 있었음
  → 검사 후 바로 출력하는 대신 결과를 저장할 flag 변수(valid)를 두어 상황에 따라 valid 값만 바꿔둔 뒤, 맨 마지막에 최종 결과를 출력하도록 수정

 

최종 코드

import sys
input = sys.stdin.readline
T = int(input())
for i in range(T):
	string = input().rstrip()
	stack = []
	valid = True
	for j in string:
		if j == '(':
			stack.append(j)
		else:
			if stack:
				stack.pop()
			else:
				valid = False
				break
	if stack:
		valid = False
	print('YES') if valid else print('NO')

메모리: 31256 KB / 시간: 44 ms

시간 복잡도: O(n*T) - 괄호 문자열의 길이만큼 * 테스트케이스의 수만큼 반복

공간 복잡도: O(n) - 하나의 스택 사용