[백준(BOJ)] 9935 문자열 폭발 - 스위프트(Swift)

[Gold IV] 문자열 폭발 - 9935

문제 링크

성능 요약

메모리: 94712 KB, 시간: 396 ms

분류

자료 구조, 스택, 문자열

제출 일자

2023년 6월 19일 08:52:36

문제 설명

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

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

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

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

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

입력

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

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

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

출력

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

 

풀이

해당 문제는 '스택'으로 접근을 하면 좋을 것이라고 생각하였고, 만약 폭발 문자열의 마지막 단어가 스택에 추가되면 검사하여 폭발 여부를 확인하고자 하였습니다.

 

만약 폭발 문자열의 마지막 단어가 스택의 마지막과 같다면 폭발 문자열의 길이만큼 뒷부분을 확인하여, 만약 폭발 문자열이라면 해당 길이만큼 스택에서 제거해주었습니다.

 

그림으로 보면 다음과 같습니다.

 

다음은 코드입니다.

let input = readLine()!
let bomb = readLine()!
var stack = [Character]()

for i in input {
    stack.append(i)

    if i == bomb.last! {    // 들어온 값이 폭탄의 마지막 값과 같을 때
        let count = stack.count

        if count >= bomb.count && String(stack[(count - bomb.count)...]) == bomb {  // 폭탄 문자열일 때
            for _ in 0..<bomb.count {   // 폭탄 문자열 길이만큼 제거
                stack.removeLast()
            }
        }
    }
}

// MARK: - String(배열)을 통해 출력 가능
print(stack.isEmpty ? "FRULA": String(stack))