[프로그래머스] - 가장 가까운 같은 글자(Kotlin)

https://school.programmers.co.kr/learn/courses/30/lessons/142086

프로그래머스

문제

문자열 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

class Solution {
    fun solution(s: String): IntArray {
        val answer = IntArray(s.length) { -1 }
        s.forEachIndexed { index, c ->
            for ( i in index -1 downTo 0){
                if(s[i] == c){
                    answer[index] = index - i
                    break
                }
            }
        }
        return answer
    }
}

• 이전에 이문제를 풀었을 때는 뒤로 탐색을 하면 해당값이 있는지 확인했었습니다.
• 이 풀이 같은 경우 문자열이 짧을 때는 좋은 방법이지만 문자열이 길어지면 길어질수록 실행시간이 길어지면 좋지 못한 코드가 됩니다.

결과 1

---

풀이 2

class Solution {
    fun solution(s: String): IntArray {
        val result = IntArray(s.length) { -1 }
        val checkArr = IntArray('z' - 'a' + 1) { -1 }
        s.forEachIndexed { index, c ->
            val currentIdx = c - 'a'
            if (checkArr[currentIdx] != -1) {
                result[index] = index - checkArr[currentIdx]
            }
            checkArr[currentIdx] = index
        }
        return result
    }
}

• 그래서 해당문제의 조건에서 문자열은 소문자만 나온 다했으므로 그 크기만큼의 배열을 만들어 해당문자를 체크 하면 더욱 최적화된 코드를 짤 수 있습니다.

결과 2