프로그래머스 가장 가까운 같은 글자 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
지문과 같습니다.

입출력 예 #2
설명 생략

풀이

간단한 구현 문제.

문자가 나온 인덱스를 저장할 charPositions 배열의 요소를 -1로 초기화한다.

charPositions는 IntArray이고 ch - 'a' 값을 인덱스로 사용하면 'a' ~ 'z'로 크기가 26인 배열을 만들면 된다.

그리고 s를 돌면서 앞에 같은 문자가 나오지 않았으면 (charPositions값이 -1이면) -1을 출력하고 앞에 같은 문자가 나왔다면 현재 인덱스에 charPositions값을 빼주면 된다. 그리고 현재 나온 문자의 인덱스 값을 경신한다.

코드

class Solution {
    fun solution(s: String): IntArray {
        val charPositions = IntArray(26){-1}
        return IntArray(s.length){
            val ret = if(charPositions[s[it]-'a'] == -1) -1 else it-charPositions[s[it]-'a']
            charPositions[s[it]-'a'] = it
            ret
        }
    }
}