[Java | 스택 | 9935번 | 골4] 문자열 폭발

☑️ 문제

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

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

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

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

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

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

 

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

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

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

 

출력

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

 

예제 입력1

mirkovC4nizCC44
C4

 

예제 출력1

mirkovniz

 

예제 입력2

12ab112ab2ab
12ab

 

예제 출력2

FRULA

 

☄️ 정답코드

import java.*;
import java.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String str = br.readLine();
        String bomb = br.readLine();
        int bombLen = bomb.length();

        Stack<Character> stack = new Stack<>();

        for (char c : str.toCharArray()) {
            stack.push(c);

            if(stack.size() >= bombLen) {
                boolean isBomb = true;

                for (int i = 0; i < bombLen; i++) {
                    if(stack.get(stack.size() - bombLen + i) != bomb.charAt(i)) {
                        isBomb = false;
                        break;
                    }
                }

                if(isBomb) {
                    for (int i = 0; i < bombLen; i++) {
                        stack.pop();
                    }
                }
            }
        }

        StringBuilder sb = new StringBuilder();
        for (char c: stack) {
            sb.append(c);
        }

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

 

🪐 풀이

스택이라는 자료구조를 활용해 풀면 되는 문제이다!

 

문자열과 관련된 문제들 중 스택과 관련된 문제가 많은 것 같다(그냥 내 의견 ..)

 

주어진 문자열을 하나씩 스택에 넣고, 폭발 문자열인지 확인해주면 된다!

 

시간복잡도를 보자면 주어진 문자열을 하나씩 스택에 넣으면서 동작하니까 O(N)이다

N(문자열 길이)이 1,000,000이니까 O(N)이면 충분!