[Swift] lv.1 숫자 문자열과 영단어 | Programmers

문제 설명

네오와 프로도가 숫자놀이를 하고 있습니다. 네오가 프로도에게 숫자를 건넬 때 일부 자릿수를 영단어로 바꾼 카드를 건네주면 프로도는 원래 숫자를 찾는 게임입니다.

다음은 숫자의 일부 자릿수를 영단어로 바꾸는 예시입니다.

1478 → "one4seveneight"
234567 → "23four5six7"
10203 → "1zerotwozero3"
이렇게 숫자의 일부 자릿수가 영단어로 바뀌어졌거나, 혹은 바뀌지 않고 그대로인 문자열 s가 매개변수로 주어집니다. s가 의미하는 원래 숫자를 return 하도록 solution 함수를 완성해주세요.

참고로 각 숫자에 대응되는 영단어는 다음 표와 같습니다.
0 zero
1 one
2 two
3 three
4 four
5 five
6 six
7 seven
8 eight
9 nine

제한 사항

1 ≤ s의 길이 ≤ 50
s가 "zero" 또는 "0"으로 시작하는 경우는 주어지지 않습니다.
return 값이 1 이상 2,000,000,000 이하의 정수가 되는 올바른 입력만 s로 주어집니다.

1차 코드

import Foundation

func solution(_ s:String) -> Int {
    var result = ""
    var searchBar = ""

    for i in s {
        if i.isNumber {
            result += String(i)
            searchBar = ""
        } else {
            searchBar += String(i)

            switch searchBar {
            case "zero":
                result += "0"
                searchBar = ""
            case "one":
                result += "1"
                searchBar = ""
            case "two":
                result += "2"
                searchBar = ""
            case "three":
                result += "3"
                searchBar = ""
            case "four":
                result += "4"
                searchBar = ""
            case "five":
                result += "5"
                searchBar = ""
            case "six":
                result += "6"
                searchBar = ""
            case "seven":
                result += "7"
                searchBar = ""
            case "eight":
                result += "8"
                searchBar = ""
            case "nine":
                result += "9"
                searchBar = ""
            default:
                break
            }
        }
    }
    return Int(result)!
}

1차 결과

보완점

우선 서치바라는 내가 만든 개념으로 접근한 일이 매우 뿌듯하긴하다.
그런데 불필요하게 반복되는 코드가 너무 많은 것이 문제이다.
그리고 일단 케이스가 너무 많고 길다. 나중에 엄청나게 많은 케이스를 다루는 일을 하게된다면 다른 방법이 뭐가 있는지
다른 사람의 풀이를 통해서 배워볼 필요가 있다.

다른 사람의 풀이

import Foundation

func solution(_ s:String) -> Int {
    // 모든 경우의 숫자를 저장
    let arr = ["zero","one","two","three","four","five","six","seven","eight","nine"]
    var str = s
    for i in 0..<arr.count {
        // arr안에 있는 zero부터 nine까지 있는지 검사 후 변경
        // 이해하기 쉽게 arr[0] = "zero"이다 zero는 0으로 변환해줘야한다
        // 인덱스 값으로 변환해주면 된다 arr[1]도 "one"이기때문에
        str = str.replacingOccurrences(of: arr[i], with: String(i))
    }
    return Int(str)!
}

이렇게 풀면 더 깔끔하게 여러 경우의 수를 다룰 수 있을 것 같다.
처음보는 고차함수가 하나 나왔는데
replacingOccurrences(of:with:) 메서드는 문자열에서 지정된 문자열 또는 문자를 다른 문자열 또는 문자로 바꾸는 데 사용됩니다. 이 메서드는 문자열을 변경하지 않고 새로운 문자열을 반환합니다.
이러한 기능이다.