BOJ-9935 문자열 폭발 Swift

문제

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

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

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

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

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

입력

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

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

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

출력

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

내가 푼 풀이

접근방법: 1. Swift에 내장된 문자열 메서드(시간초과)

 

스택을 사용하면 될꺼라 생각했지만, 더 간단하게 Swift에서 제공하는 메서드를 이용하면 되지 않을까 했다.

import Foundation
// 입력받기
var str = readLine()!
let bomb = readLine()!

// 폭발 문자열 포함이 되었다면 해당 문자열을 빈 문자열로 치환
while str.contains(bomb) {
    str = str.replacingOccurrences(of: bomb, with: "")
}
// 모두 바꾸고, 문자열이 비어있다면 FRULA, 아니라면 문자열 출력
if str.isEmpty {
    print("FRULA")
} else {
    print(str)
}

 replacingOccurrences 메서드는 해당 문자열을 다른 문자열로 치환해준다.

코드도 간결해서 바로 통과될 줄 알았지만, 해당 문자열에 포함되었는지 확인하는 contains는 O(N), 

replacingOccurrences는 보기에 문자를 바로 치환해주지만, 주어진 문자열을 모두 검사하고, 치환하는 과정을 거친다 O(N)

따라서 위 코드의 시간복잡도는 아마 O($N^{2}$)과 근접한 시간이 걸린다.. 그래서 44퍼에서 시간초과가 났다.

 

 

접근방법 2: 스택을 사용

주어진 문자열을 순회하며, 스택에 push하고,

스택에 상단문자부터 폭발문자열의 길이만큼 뽑아낸 문자열이 폭발문자열인경우 모두 pop해준다.

스택이 비어있다면 FRULA, 비어있지않다면 문자열로 출력

 

import Foundation

// 입력받기
var leftStack = Array(readLine()!)
let bomb = Array(readLine()!)
let bCount = bomb.count
var rightStack = [Character]()

// 스택에 넣기
while !leftStack.isEmpty {
    rightStack.append(leftStack.removeLast())
    // 스택이 폭발 문자열길이만큼 넣어졌다면 검사
    if rightStack.count >= bCount {
        check()
    }
}
// 스택의 상단원소부터 폭발 문자열 길이만큼의 문자를 뽑아서, 해당 문자와 폭발 문자열이 같은 경우 pop
func check() {
    var s = [Character]()
    for i in (rightStack.count-bCount..<rightStack.count).reversed() {
        s.append(rightStack[i])
    }
    if s == bomb {
        for i in 0..<s.count {
            rightStack.removeLast()
        }
    }
}

if rightStack.isEmpty {
    print("FRULA")
} else {
    print(String(rightStack.reversed()))
}