문제
상근이는 문자열에 폭발 문자열을 심어 놓았다. 폭발 문자열이 폭발하면 그 문자는 문자열에서 사라지며, 남은 문자열은 합쳐지게 된다.
폭발은 다음과 같은 과정으로 진행된다.
- 문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다.
- 새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다.
- 폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다.
상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 "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()))
}'코딩테스트 > 백준' 카테고리의 다른 글
| BOJ-오아시스 재결합 Swift (0) | 2024.05.10 |
|---|---|
| BOJ-17299 오등큰수 Swift (0) | 2024.05.10 |
| BOJ-11444 피보나치 수 6 Swift (0) | 2024.05.05 |
| BOJ-10830 행렬 제곱 Swift (0) | 2024.05.05 |
| BOJ-11401 이항 계수 3 Swift (0) | 2024.05.03 |
