[프로그래머스 LV.0] 컨트롤 제트

컨트롤 제트

문제 설명

숫자와 "Z"가 공백으로 구분되어 담긴 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이때 "Z"가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 "Z"로 이루어진 문자열 s가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해 보세요.

제한 사항

• 1 ≤ s의 길이 ≤ 200
• -1,000 < s의 원소 중 숫자 < 1,000
• s는 숫자, "Z", 공백으로 이루어져 있습니다.
• s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.
• 연속된 공백은 주어지지 않습니다.
• 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
• s는 "Z"로 시작하지 않습니다.
• s의 시작과 끝에는 공백이 없습니다.
• "Z"가 연속해서 나오는 경우는 없습니다.

입출력 예

s	result
"1 2 Z 3"	4
"10 20 30 40"	100
"10 Z 20 Z 1"	1
"10 Z 20 Z"	0
"-1 -2 -3 Z"	-3

입출력 예 설명

입출력 예 #1

• 2 다음 Z 이기 때문에 2가 지워지고, 1 + 3 = 4를 return 합니다.

입출력 예 #2

• 10 + 20 + 30 + 40 = 100을 return 합니다.

입출력 예 #3

• "10 Z 20 Z 1"에서 10 다음 Z, 20 다음 Z로 10, 20이 지워지고 1만 더하여 1을 return 합니다.

입출력 예 #4, #5

설명 생략

---

제출

import Foundation

func solution(_ s:String) -> Int {
    var s = s.split(separator: " ").map{String($0)}
    
    while s.contains("Z"){
        var index = s.firstIndex(of: "Z")!
        if index - 1 >= 0 {
            s.remove(at: index)
            s.remove(at: index - 1)
        }else{
            s.remove(at: index)
        }
    }

    return s.map{Int($0)!}.reduce(0, +)
}

입력 문자열을 공백기준으로 나눈 후 String 배열 s로 생성
s에 Z가 없을 때까지 반복하며, Z가 처음으로 등장하는 인덱스를 구한 후 인덱스-1이 0보다 크면 Z와 Z앞의 요소를 지운다
인덱스-1이 0보다 작으면 Z만 지운다

이후 s를 map 함수를 사용하여 Int형 배열로 바꾼 후, reduce함수를 사용하여 합을 구한다