[백준/Java] 4949 균형잡힌 세상(Stack, replaceAll())

https://www.acmicpc.net/problem/4949

📝 풀이 2 : Stack (170ms)

`Stack`을 사용하여 여는 괄호를 저장하고, 닫는 괄호를 만나면 스택의 맨 위 항목과 짝을 맞춰 균형을 확인한다.

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

public class Main {
    // 각 줄마다 해당 문자열이 균형을 이루고 있으면 "yes"를, 아니면 "no"를 출력
    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        while (true) {
            /* [1] 입력 */
            String str = br.readLine(); // 영문, 공백, 소괄호, 대괄호로 이루어짐(100글자보다 작거나 같음)
            if (str.equals(".")) break; // 온점으로 끝남
            /* [2] 짝 맞는 괄호 판단 */
            Stack<Character> stack = new Stack<>(); //✨
            boolean isBalanced = true; // 플래그
            for (char c : str.toCharArray()) {
                if (c == '[' || c == '(') stack.push(c); // 여는 괄호 push✨
                else if ((c == ']') || (c == ')')) { // 닫는 괄호
                    if (stack.isEmpty() || ((c == ']' && stack.peek() != '[')
                            || (c == ')' && stack.peek() != '('))) { // 스택이 비어있거나 짝이 맞지 않으면
                        isBalanced = false; // 불균형 1
                        break;
                    }
                    stack.pop(); // 짝이 맞는 경우 스택에서 제거✨
                }
            }

            if (!stack.isEmpty()) isBalanced = false; // 불균형 2
            sb.append(isBalanced ? "yes" : "no").append("\n");
        }

        /* [3] 출력 */
        System.out.println(sb.toString());
    }
}

📝 풀이 1 : contains() + replaceAll() (476ms)

모든 괄호 이외의 문자를 제거한 후, 괄호 쌍을 반복적으로 제거하여 남은 문자열이 없는지를 확인한다.

import java.io.*;

public class Main {
    // 각 줄마다 해당 문자열이 균형을 이루고 있으면 "yes"를, 아니면 "no"를 출력
    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        while (true) {
            /* [1] 입력 */
            String str = br.readLine(); // 영문, 공백, 소괄호, 대괄호로 이루어짐(100글자보다 작거나 같음)
            if (str.equals(".")) break; // 온점으로 끝남
            /* [2] 짝 맞는 괄호 제거 */
            str = str.replaceAll("[^\\[\\]\\(\\)]", ""); // 괄호가 아닌 문자 제거
            while (str.contains("()") || str.contains("[]")) {
                str = str.replaceAll("(\\[\\])|(\\(\\))", ""); // 안쪽부터 짝이 맞는 괄호 제거✨
            }
            sb.append((str.isBlank() ? "yes" : "no") + "\n");
        }

        /* [3] 출력 */
        System.out.println(sb.toString());
    }
}