9953번: 문자열 폭발 - Kotlin (스택)

9935번: 문자열 폭발

문제

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

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

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

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

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

입력

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

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

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

출력

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

 

나의 풀이

 이 문제를 처음 읽었을 때 든 생각은 너무 단순하다는 것이었다. 골드4의 문제치고 너무 단순하다는 생각이 들었다. 아니나 다를까 문자열의 최대 크기가 1000000이었다. 이렇게 되면 단순하게 replace를 사용했을 때, 계속해서 최대 길이의 문자열을 만들어내야 함으로 메모리 초과가 발생하였다. 그래서 Stack을 이용해서 n번만 반복하게 하고 매번 폭탄 문자열의 길이만큼 스택의 위에서부터 확인해서 폭탄과 일치하면 폭탄의 길이만큼 pop을 시켜주었다.

 

import java.lang.StringBuilder
import java.util.LinkedList
import java.util.Queue
import java.util.Stack

fun main() = with(System.`in`.bufferedReader()) {
    val input = readLine().toString()
    val bomb = readLine().toString()

    val bombLength = bomb.length
    val stack : Stack<String> = Stack()

    repeat(input.length) {
        stack.add(input[it].toString())

        if (stack.size >= bombLength) {
            var check = true

            for (i in 0 until bombLength) {
                if (stack[stack.size - bombLength + i] != bomb[i].toString()) {
                    check = false
                    break
                }
            }

            if (check) {
                repeat(bombLength) {
                    stack.pop()
                }
            }
        }
    }

    val sb = StringBuilder()

    if (stack.isEmpty())
        println("FRULA")
    else {
        sb.append(stack.joinToString(""))
        println(sb)
    }

}