Algorithm/Baekjoon

백준 / 9935번 / 문자열 폭발

Gyuri 2022. 12. 31. 12:22

[문제]

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

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

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

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

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

[입력]

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

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

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

 

[출력]

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

[예제 입력]

mirkovC4nizCC44
C4

[예제 출력]

mirkovniz
 

[예제 입력 2]

12ab112ab2ab
12ab

[예제 출력 2]

FRULA

 

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

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

        String word = br.readLine(); // 문자열
        String bomb = br.readLine(); // 폭발 문자열
        int bomblen = bomb.length(); // 폭발 문자열 길이

        Stack<Character> stack = new Stack<>();
        
        for(int i=0; i<word.length(); i++) {
            char c = word.charAt(i);
            stack.add(c);

            // 폭발 문자열 길이와 같거나 넘은 경우
            if(stack.size() >= bomblen) {
                boolean flag = true;

                // 폭발 문자열과 같은 지 검사
                for(int j=0; j<bomblen; j++) {
                    char c1 = stack.get(stack.size() - bomblen + j);
                    char c2 = bomb.charAt(j); 

                    if(c1 != c2) {
                        flag = false;
                        break;
                    }
                }

                // 폭발 문자열과 같은 경우 (단어 폭파시키기)
                if(flag) {
                    for(int k=0; k<bomblen; k++) {
                        stack.pop();
                    }
                }
            }
        }

        if(stack.size() == 0) {
            System.out.println("FRULA");
        }
        else {
            StringBuilder sb = new StringBuilder();
            for (Character c : stack) {
                sb.append(c);
            }
            System.out.println(sb);
        }
    }
}

문자열과 폭발 문자열이 주어지고, 폭발 문자열이 있으면 그 문자열을 없애고 남은 문자열에서 폭발 문자열이 없을 때 까지 반복하는 형식이다. 

 

Stack을 이용해서 풀었는데, 모든 문자를 Stack에 넣은 후 검사하는 것이 아니라

한 글자씩 추출해서 Stack에 넣고 폭발 문자열의 길이(bomblen) 씩만 검사를 하는 형식으로 문제를 풀었다.

 

boolean 변수를 하나 두고,  bomblen만큼 글자를 검사하면

폭발 문자열의 글자와 내 문자열이 일치하는지 알 수 있다.

 

일치하는 경우엔 또 for문을 돌며 폭발 문자열 길이만큼 stack.pop 을 통해 문자열을 없애주면 된다.

 

이 과정이 끝난 후, 스택에 남아있는 문자가 없다면 FRULA를 출력하고,

그렇지 않은 경우 (남아있는 문자가 있는 경우)엔 해당 문자열을 다 이어서 출력해주면 된다.

이 때, StringBuilder를 쓰지 않고 표준 출력을 쓰면 시간초과가 뜨기 때문에 StringBuilder를 써줬다.

 

다른 사람들 코드를 봤을 때, 애초에 Stack을 쓰지 않고 StringBuilder를 사용하기도 했다.