[Java] 백준 9935 - 문자열 폭발

Baekjoon_9935 - 문자열 폭발

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

문제

상근이는 문자열에 폭발 문자열을 심어 놓았다. 폭발 문자열이 폭발하면 그 문자는 문자열에서 사라지며, 남은 문자열은 합쳐지게 된다.

폭발은 다음과 같은 과정으로 진행된다.

• 문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다.
• 새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다.
• 폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다.

상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 "FRULA"를 출력한다.

폭발 문자열은 같은 문자를 두 개 이상 포함하지 않는다.

입력

첫째 줄에 문자열이 주어진다. 문자열의 길이는 1보다 크거나 같고, 1,000,000보다 작거나 같다.

둘째 줄에 폭발 문자열이 주어진다. 길이는 1보다 크거나 같고, 36보다 작거나 같다.

두 문자열은 모두 알파벳 소문자와 대문자, 숫자 0, 1, ..., 9로만 이루어져 있다.

출력

첫째 줄에 모든 폭발이 끝난 후 남은 문자열을 출력한다.

---

해결방법

• 처음 이 문제를 읽었을 때, 골드5 수준치고는 어렵지 않다고 생각했다.
  • contains() 함수를 이용해 폭발 문자열이 포함되어 있다면 replace(boom, "") 로 대체하는 것을 반복하면 된다고 생각했다.
• 그래서 아래 첫 번째 코드처럼 풀었더니 메모리 초과가 났다.
  • 아마 replace() 함수 내부에서 메모리를 너무 많이 잡아먹는 것 같았다.
• 따라서 다른 방법을 생각해봐야 했다. 그리고 스택을 이용해 문제를 해결해보기로 했다.
  • 한글자씩 스택에 쌓아가며 boolean 타입의 isBoom 변수가 true라면 바로 pop() 해버리는 것이다.
  • 폭발 문자열 판별하는 방법
    • 스택에 쌓인 문자열의 길이가 폭발 문자열과 같아지는 순간 탐색을 시작한다.
    • 스택의 길이 - 폭발 문자열 길이 부터 스택의 끝 까지를 폭발 문자열과 비교하며, 같으면 폭발 문자열이라는 뜻.
  • 최종적으로 스택에 남은 글자들을 모두 StringBuilder 에 쌓고 출력한다.

코드

// 메모리 초과 난 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Baekjoon_9935 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String word = br.readLine();
        String boom = br.readLine();

        String result = getResult(word, boom);

        if(result.equals("")){
            System.out.println("FRULA");
        }else{
            System.out.println(result);
        }
    }

    public static String getResult(String word, String boom) {
        if (word.contains(boom)) {
            word = word.replace(boom, "");
            word = getResult(word, boom);
        }
        return word;
    }
}

// 정답 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Baekjoon_9935 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Stack<Character> st = new Stack<>();

        String word = br.readLine();
        String boom = br.readLine();

        for (int i = 0; i < word.length(); i++) {
            st.push(word.charAt(i));

            if (st.size() >= boom.length()) {
                boolean isBoom = true;

                for (int j = 0; j < boom.length(); j++) {
                    if (st.get(st.size() - boom.length() + j) != boom.charAt(j)) {
                        isBoom = false;
                        break;
                    }
                }
                if (isBoom) {
                    for (int j = 0; j < boom.length(); j++) {
                        st.pop();
                    }
                }
            }

        }

        StringBuilder sb = new StringBuilder();
        for (Character data : st) {
            sb.append(data);
        }

        System.out.println(sb.length() == 0 ? "FRULA" : sb.toString());
    }
}