[ BOJ ] 9012 : 괄호 ( SILVER 4 ) / C, Python

문제

괄호 문자열(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”를 한 줄에 하나씩 차례대로 출력해야 한다. 

풀이 과정

여는 괄호로 시작해서, 닫는 괄호로 정확히 닫아야 올바른 괄호 문자열이기에, 스택을 이용해 여는 괄호를 스택에 저장하고, 닫는 괄호가 나오면 스택에서 여는 괄호를 역순으로 빼서 연결한다.

 

Ex. ((())())

Idx 0  (  : 여는 괄호로, 스택에 저장한다. -> Stack : [ ( ]

Idx 1  (  : 여는 괄호로, 스택에 저장한다. -> Stack : [ (  ( ]

Idx 2  (  : 여는 괄호로, 스택에 저장한다. -> Stack : [ (  (  ( ]

Idx 3  )  : 닫는 괄호로, 마지막에 열었던 괄호와 연결되기에, 스택에 가장 나중에 넣은 여는 괄호를 뺀다. -> Stack : [ (  ( ]

이 닫는 괄호로 인하여, 현재 완성된 괄호는 ((())()) 이다.

Idx 4  )  : 닫는 괄호로, 마지막에 열었던 괄호와 연결되기에, 스택에 가장 나중에 넣은 여는 괄호를 뺀다. -> Stack : [ ( ]

이 닫는 괄호로 인하여, 현재 완성된 괄호는 ((())()) 이다.

Idx 5  (  : 여는 괄호로, 스택에 저장한다. -> Stack : [ (  ( ]

Idx 6  )  : 닫는 괄호로, 마지막에 열었던 괄호와 연결되기에, 스택에 가장 나중에 넣은 여는 괄호를 뺀다. -> Stack : [ ( ]

이 닫는 괄호로 인하여, 현재 완성된 괄호는 ((())()) 이다.

Idx 7  )  : 닫는 괄호로, 마지막에 열었던 괄호와 연결되기에, 스택에 가장 나중에 넣은 여는 괄호를 뺀다. -> Stack : [ ]

이 닫는 괄호로 인하여, 현재 완성된 괄호는 ((())()) 이다.

여는 괄호가 닫는 괄호가 전부 매치되서 스택에 남아있는 여는 괄호가 없으며, 스택이 비어있을 때 닫는 괄호가 나오지 않았느니 이는 올바른 괄호 문자열이다. 

 

다음은 올바르지 않은 문자열이다.

1. 스택이 비어있는데 닫는 괄호가 나온다. [ (()))) ]       [ ))(()) ]

2. 스택에 여는 괄호가 남아있다. [ (())(( ]        [ (((()) ]

 

여는 괄호가 추가되거나 삭제되는 연산만 진행해주면 되니, 여는 괄호를 +1, 닫는 괄호를 -1로 치환해서 연산을 진행해도 무방하다.

C

스택 풀이 코드

#include <stdio.h>
#define SIZE 55

int stack[SIZE];
int stack_num = 0;

int main(void) {
	int test;
	scanf("%d", &test);
	
	for(int i = 0; i < test; i++) {
		char str[100];
		scanf("%s", str);
		int idx = 0;
		int done = 0;
		while(str[idx] != 0) {
			if(str[idx] == '(') stack[stack_num++] = 1;
			else {
				if(stack_num == 0) { // 여는 괄호가 없는데 등장한 닫는 괄호
					printf("NO\n");
					done = 1;
					break;
				}
				else stack_num -= 1;
			}
			
			idx += 1;
		}
		
		if(done == 0) {
			if(stack_num != 0) printf("NO\n"); // 닫히지 않은 괄호
			else printf("YES\n");
		}
	
		stack_num = 0; // 초기화
	}
	
	return 0;
}

 

연산 풀이 코드

#include <stdio.h>
#include <string.h>

int main(void) {
    int t;
    scanf("%d", &t);
    while (t--) {
        char str[55];
        scanf("%s", str);
        
        int stack = 0;
        int done = 1;
        for (int i = 0; i < strlen(str); i++) {
            if (str[i] == '(') stack += 1;
            else {
                if (stack == 0) {
                    done = 0;
                    break;
                }
                else stack -= 1;
            }
        }

        if (stack == 0 && done == 1) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}

Python

import sys
input = sys.stdin.readline

t = int(input().rstrip())
for _ in range(t):
    s = input().rstrip()

    stack = 0
    done = 1
    for i in range(len(s)):
        if s[i] == '(': stack += 1
        else:
            if stack == 0:
                done = 0
                break
            stack -= 1
    
    print('YES' if stack == 0 and done == 1 else 'NO')