[Swift] lv.1 가장 가까운 같은 글자 | Programmers

문제 설명

문자열 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이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.

제한 사항

1 ≤ s의 길이 ≤ 10,000
s은 영어 소문자로만 이루어져 있습니다.

1차 코드

import Foundation

func solution(_ s:String) -> [Int] {
    var result:[Int] = []
    let elements = s.map { String($0) }

    for i in elements.count {
        var idx = i
        var isTrue = false

        while idx > 0 {
            idx -= 1
            if elements[idx] == elements[i] {
                result.append(i - idx)
                isTrue = true
                break
            }
        }
        if isTrue == false {
            result.append(-1)
        }
    }
    return result
}

1차 결과

1차 수정

elements.count를 for문에 그대로 돌리면 시퀀스가 맞지 않는데 횟수를 지정한다는걸 실수로 count 자체를 for문을 돌린 것 같다.
0..<을 추가해서 0부터 카운트만큼 for문을 차례대로 돌리면 될 것 같다.

2차 코드

import Foundation

func solution(_ s:String) -> [Int] {
    var result:[Int] = []
    let elements = s.map { String($0) }

    for i in 0..<elements.count {
        var idx = i
        var isTrue = false

        while idx > 0 {
            idx -= 1
            if elements[idx] == elements[i] {
                result.append(i - idx)
                isTrue = true
                break
            }
        }
        if isTrue == false {
            result.append(-1)
        }
    }
    return result
}

2차 결과