[Swift 알고리즘] 백준 9935 (문자열 폭발)

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

 

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

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

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

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

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

입력

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

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

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

출력

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

 

이미지를 설명해보세요

 

해결 방법

Stack을 통해 구현하면 됩니다. stack에 input문자들을 넣는다고 생각하게 된다면, stack에는 ["a", "b", "c"...]이런식으로 쌓일 것 입니다. 여기서 현재 넣는 문자열과 폭발시키는 문자열의 마지막을 비교합니다. 그리고 나서 뒷 문자열을 비교군에 알맞게 자른 후 만약 폭발시키는 문자열과 같다면 pop을 통해 문자열을 갱신시키면 됩니다.

 

func sol3() {
    let input = String(readLine()!)
    let criteria = String(readLine()!)
    let criteriaCount = criteria.count
    var stack:[Character] = []
    for i in input {
        stack.append(i)
        if i == criteria.last {
            if stack.count >= criteriaCount && String(stack[(stack.count - criteriaCount)...]) == criteria {
                for _ in 0..<criteriaCount {
                    stack.popLast()
                }
            }
        }
    }
    print(stack.isEmpty ? "FRULA" : String(stack))
}

sol3()

 

실패 과정

1. 반복문 순회 (시간초과)

 - replacingOccurrences(of, with)를 통해 풀려고 했지만 시간초과가 났습니다.

import Foundation

func sol() {
    let input = String(readLine()!)
    let criteria = String(readLine()!)
    var targetString = input
    
    while true {
        let length = targetString.count
        targetString = targetString.replacingOccurrences(of: criteria, with: "")
        if length == targetString.count {
            break
        }
    }
    if targetString == "" {
        print("FRULA")
        return
    }
    print(targetString)
}

sol()

 

 

2. String을 통해 접근 (시간초과)

- Array()를 만드는 시간이 너무 길어서 시간 초과

func sol2() {
    let input = String(readLine()!).map{String($0)}
    let criteria = String(readLine()!).map{String($0)}
    let criteriaCount = criteria.count
    var stack:[String] = []
    
    for i in input {
        stack.append(i)
        if i == criteria.last! && stack.count >= criteriaCount {
            if Array(stack[(stack.count - criteriaCount)...]) == criteria {
                stack = Array(stack[...(stack.count - criteriaCount - 1)])
            }
        }
    }
    stack.isEmpty ? print("FRULA") : print(stack.joined())
    
}

sol2()

 

3. Character를 통해 접근 (시간초과)

 - Character 또한 Array()를 만드는 시간이 너무 길어서 시간 초과

 

func sol3() {
    let input = String(readLine()!)
    let criteria = String(readLine()!)
    let criteriaCount = criteria.count
    var stack:[Character] = []
    for i in input {
        stack.append(i)
        if i == criteria.last {
            if stack.count >= criteriaCount && String(stack[(stack.count - criteriaCount)...]) == criteria {
                for _ in 0..<criteriaCount {
                    stack.popLast()
                }
            }
        }
    }
    print(stack.isEmpty ? "FRULA" : String(stack))
}

sol3()