코딩테스트

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

깨비도 2024. 6. 15. 12:28

가장 가까운 같은 글자

문자열 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을 완성해주세요.

 

💡 풀이

class Solution {
    fun solution(s: String): IntArray {
        val prevIndexMap = hashMapOf<Char, Int>()
        
        var answer = s.mapIndexed{ index, item ->
            val prevIndex = prevIndexMap[item]
            prevIndexMap[item] = index

            if (prevIndex == null) -1 else index - prevIndex
        }.toIntArray()
        
        return answer
    }
}
  1. 빈 해시맵 (Map) 을 선언한다. 이 해시맵은 Key와 value로 생성되는데 value 값은 null로 되어있다.
  2. 주어진 String s를 mapIndexed로 순회한다.
  3. prevIndex에 prevIndexMap[item]을 할당해준 뒤,
  4. prevIndexMap[item]을 index로 교체하여 값을 넣어준다.
  5. 만약 prevIndex가 null이면 앞에 값이 없다는 뜻이므로 -1을 넣어주고 그게 아니라면 앞에 값이 있다는 뜻이므로 index에서 prevIndex를 빼주면 된다.
  6. map 메소드는 List를 반환하므로 toIntArray()를 변환해주면 된다.

콘솔을 찍어보면 이러한 결과가 나오는데,

앞의 값은 prevIndex값이고 뒤의 값은 prevIndexMap[item]에 index를 할당해준 뒤의 값이다.

따라서 index - prevIndex 했을 때 몇 번째 칸 앞에 같은 글자가 있는지 알 수 있다.


다른 사람의 풀이를 보고 푼 것이다. 혼자서는 못 풀 것 같은데 언제 쯤 잘 하게 될지 모르겠다...ㅠㅠ