[swift] 숫자 문자열과 영단어(feat. replacingOccurrences)

문제

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

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

• 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로 주어집니다.

입출력 예

s	result
"one4seveneight"	1478
"23four5six7"	234567
"2three45sixseven"	234567
"123"	123

과정

1. 바꿀 문자열에 대한 숫자를 준비(배열, 딕셔너리, 열거형 등)

2. 바꿀 문자열을 찾아, 숫자로 대체한다. (대체된 결과물은 String, Array 등이 될 수 있다.)

3. 변환한 결과물을 Int로 반환한다.

답안

import Foundation

func solution(_ s:String) -> Int {
    var result = s
    let alphabetNumber = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
    // 주어진 문제에 따라서 숫자에 대응하는 영어 단어를 배열로 생성. 각 String은 index에 대응하여 대체.
    
    for i in 0..<alphabetNumber.count {
        result = result.replacingOccurrences(of: alphabetNumber[i], with: String(i))
        // replacingOccurrence(of:with:) 메서드로 주어진 문자열 안의 "zero"..."nine"까지의 문자열을 찾고(alphabetNumber[i]),
        // 찾은 문자열(alphabetNumber[i])는 index인 i로 대체된다. 해당 메소드의 반환하는 타입은 String이므로 i를 String으로 형변환.
    }
    return Int(result)!
    // 숫자로만 이루어진 String타입의 result를 Int로 반환 
}

'숫자 문자열과 영단어'에서는 주어진 문자열에서
바꿀 영단어를 찾고, 이를 숫자로 변환하는 작업이 가장 중요합니다.

이처럼 문자열의 일부를 교체하는 작업을 간단하게 수행할 수 있는
replacingOccurrences(of:with:) 메서드에 대해서 알아보겠습니다.

func replacingOccurrences(of target: String, with replacement: String) -> String

• target : 교체하고자하는 대상 문자열
• replacement : 'target'을 대체할 새로운 문자열

이 메서드는 원본 문자열에서 'target'과 일치하는 모든 부분을 'replacement'로 대체한 새로운 문자열을 반환하며, 대소문자까지도 구분할 수 있습니다.

다른 예시를 보면,

var originalString = "apple orange apple banana"
let newString = originalString.replacingOccurrences(of: "apple", with: "kiwi")
print("newString")		// "kiwi orange kiwi banana"

이처럼 해당하는 문자열("apple")만을 찾아 대체 문자열("kiwi")로 바꿔주고 있습니다.

후기

replacingOccurrences(of:with:) 메서드를 사용하지 않고 해결한 방법을 함께 알아보겠습니다.
과정은 위와 동일하나,

import Foundation

func solution(_ s: String) -> Int {
    // 각 영단어에 대응하는 숫자를 저장한 딕셔너리
    let numToStringDict: [String: String] = [
        "zero": "0", "one": "1", "two": "2", "three": "3", "four": "4",
        "five": "5", "six": "6", "seven": "7", "eight": "8", "nine": "9"
    ]

    // 문자열을 각 문자로 분리하여 배열로 저장
    let sArray = s.map {$0}			// ["2", "3", "f", "o", "u", "r", "5", "s", "i", "x", "7"]
    var letter: [String] = []		// 주어진 String 내의 알파벳들을 저장할 배열
    var result: [String] = []		// 주어진 String 내의 숫자들만 저장할 배열. 결과

    // 문자열을 하나씩 순회
    for i in sArray {
        if i.isNumber {
            // 숫자인 경우 바로 결과(result)에 추가
            result.append(String(i))
            continue
        }
        // 문자인 경우 letter에 append하고 대응되는 숫자가 있는지 확인
        // f, fo, fou, four -> 4
        // s, si, six -> 6
        letter.append(String(i))
        if let mappedNumber = numToStringDict[letter.joined()] {
            // 대응되는 숫자가 있으면 결과에 추가하고 letter 초기화
            result.append(mappedNumber)
            letter.removeAll()
        }
    }

    // 결과 배열을 문자열로 합치고 정수로 변환하여 반환
    return Int(result.joined())!
}

print(solution("23four5six7"))