알고리즘

[ 알고리즘 ] 스택을 이용한 VPS 판단

hanjuCoding 2024. 7. 22. 17:33

스택 쌓는 나서스

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

 

예제 입력 1

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

 

예제 출력 1

NO
NO
YES
NO
YES
NO

예제 입력 2 

3
((
))
())(()

 

예제 출력 2

NO
NO
NO

 

 

풀이

이 문제는 자주 등장하는 문제인데, 나는 그동안 스택을 이용하지 않고 괄호들을 리스트에 하나씩 넣으면서 괄호 한 쌍이 완성되면 리스트에서 제거해서 남은 문자를 보고 판단하는 방식을 사용했었다.(사실 스택 사용하는 방식이랑 비슷함..아닌가? 해석하기 나름이다) 

이번에 그냥 스택으로 하는 풀이를 정리해보고자 한다.

https://st-lab.tistory.com/178 이미지 참고

 

그림과 같이 스택에 문자가 ' ( '이면 push를하고, ' ) ' 면  pull을 해서 괄호 쌍을 제거하는 방식이다.

만약에 스택이 비었는데 ' ) '가 들어온다던가 마지막 연산을 처리한 후 스택이 비어있지 않았다면 VPS가 될 수 없다.

 

정답 코드 :)

import java.io.*;
import java.util.*;



public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(br.readLine());
        for(int i = 0; i < T; i++) {
            String s = br.readLine();
            Stack<Character> stack = new Stack<>();
            boolean result = true;
            for(int j = 0; j < s.length(); j++) {
                char c = s.charAt(j);
                if(c =='(') stack.push(c);
                else if(stack.empty()){
                    result =false;
                    break;
                }else{
                    stack.pop();
                }
            }
            System.out.println(result && stack.empty()?"YES":"NO");
        }

    }


}