[백준 9935] swift 문자열 폭발

문제 설명

 

쉽게 말해 폭탄이 있는 문자를 지워주면 된다. 또 지우고 합친 상태에서 폭탄이 있으면 또 지워주고 반복

 

풀이

이전에 프로그래머스에서 비슷한 문제를 풀어 바로 풀이법을 생각해 낼 수 있었다. (햄버거만들기)

 

1. stack에 문자열 하나씩 넣어준다.

2. 그때마다 stack에 폭탄이랑 똑같은 문자열이 있는지 체크하고 있다면 폭탄수만큼 pop 해주면 된다.

 

코드

let targetString = readLine()!.map { String($0) }
let bomb = readLine()!.map { String($0) }
var stack = [String]()

for i in 0..<targetString.count {
    stack.append(targetString[i])
    if stack.last! == bomb.last! // 현재 스택의 마지막 문자와 폭탄의 마지막 문자 같은지
        && stack.count >= bomb.count // 스택의 개수가 폭탄의 개수보다 많은지 (인덱스 에러나지 않게 체크)
        && stack[stack.count-bomb.count..<stack.count].joined() == bomb.joined() { // 스택에 폭탄이 있는지
        for _ in 0..<bomb.count {
            stack.removeLast()
        }
    }
}

print(stack.isEmpty ? "FRULA" : stack.joined())

 

후기

비슷한 문제를 푼 경험이 있어 비교적 쉽게 문제를 풀 수 있었다.

아직 많이 부족하다고 생각하지만.. 조금씩 성장해간다고 느낀다..!!