[프로그래머스][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이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.

제한사항

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

 

풀이

---

import Foundation

func solution(_ s:String) -> [Int] {
    var answer: [Int] = []
    
    // 문자열을 리스트로 변환 ex. "aaa" -> ["a", "a", "a"]
    let elements = s.map { String($0) }
    
    for i in 0..<elements.count {        
        var index = i // 하나씩 빼며 검사할 문자 인덱스
        var isExist = false // 같은 글자가 존재하는지
        
        // 하나씩 앞으로 가며 검사
        while (index > 0) {
            index -= 1
            
            if elements[index] == elements[i] {
                answer.append(i - index) // 현재 인덱스 - 검사 인덱스 = 몇번째 앞인지
                isExist = true
                break
            }
        }
        
        // 앞쪽에 같은 글자가 없으면 -1
        if isExist == false {
            answer.append(-1)
        }
    }
    
    return answer
}

 

해설

처음엔 String.index를 구해서 s에 바로 접근했는데 String.index를 구하는게 내부적으로 시간이 많이 걸리는지 시간초과가 떴습니다.

그래서 빠르게 접근할 수 있도록 문자열 요소를 하나씩 리스트에 담아 배열 인덱스로 접근하여 해결했습니다.