프로그래머스 Lv.1 Swift 가장 가까운 글자

 

문제 

문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.
예를 들어, s="banana"라고 할 때,  각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.

• b는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
• a는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
• n은 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
• a는 자신보다 두 칸 앞에 a가 있습니다. 이는 2로 표현합니다.
• n도 자신보다 두 칸 앞에 n이 있습니다. 이는 2로 표현합니다.
• a는 자신보다 두 칸, 네 칸 앞에 a가 있습니다. 이 중 가까운 것은 두 칸 앞이고, 이는 2로 표현합니다.

따라서 최종 결과물은 [-1, -1, -1, 2, 2, 2]가 됩니다.

 

 

입출력 예

s	result
"banana"	[-1, -1, -1, 2, 2, 2]
"foobar"	[-1, -1, 1, -1, -1, -1]

 

 

풀이

func solution(_ s:String) -> [Int] {
    var answer: [Int] = []
    var charDict: [Character:Int] = [:] // s의 문자열을 하나씩 인덱스 순서와 함께 저장
    let charArr = s.map { $0 }
    var currIndex = 0

    for char in charArr {
        if charDict[char] == nil {
            answer.append(-1)
        } else {
            let index = currIndex - charDict[char]!
            answer.append(index)
        }

        // 기존에 앞에서 나온 charDict[char]에 현재 인덱스 넘버를 덮어씌움.
        charDict[char] = currIndex
        currIndex += 1
    }
    return answer
}

 

일단 다른 분들의 풀이도 보면서 했는데 비슷하더라도 과정을 이해하려 노력했다.

enumerated를 사용한 방법도 있었는데 그에 관한 것을 완전히 이해했다고 하기에는 부족해서 더 공부해봐야겠다.

 

 

-> enumerated를 사용한 풀이

func solution(_ s:String) -> [Int] {
    var charDict: [Character:Int] = [:]
    var answer: [Int] = []

	// EnumeratedSequence<String>
    s.enumerated().forEach { (index, char) in
        if let previousCharacterIndex = charDict[char] {
            charDict[char] = index
            answer += [index - previousCharacterIndex]

        } else {
            answer.append(-1)
            charDict[char] = index
        }
    }
    return answer
}