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

문제

문자열 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. map을 만들어서 key에는 알파벳을, value에는 해당 알파벳의 가장 마지막 인덱스를 저장한다.

var map = HashMap<Char, Int>()

 

2. 주어진 문자열 s의 길이만큼 반복하며 문자열을 문자로 쪼갠다.

for(i in 0 until s.length){
    var c = s[i]

 

3. map.getOrDefault를 사용해 map에 알파벳이 있다면 c를 반환하고 없다면 -1을 반환하도록 한다.

아래 코드는 map에 해당 알파벳이 없을 경우이다. 이 경우 map에 현재 인덱스 번호를 넣어주고,

알파벳이 처음 나왔기 때문에 answer에 -1을 추가한다.

if(map.getOrDefault(c, -1) == -1){
    map[c] = i
    answer[i] = -1
}

 

4. 만약 map에 해당 알파벳이 있다면 현재 인덱스에서 마지막 인덱스 값을 뺀 값을 answer에 추가한다.

그리고 현재 알파벳의 마지막 인덱스 값을 바꿔준다.

else{
    answer[i] = i - map[c]!!
    map[c] = i

 

코드

class Solution {
    fun solution(s: String): IntArray {
        var answer = IntArray(s.length, {0})
        var map = HashMap<Char, Int>()
        
        for(i in 0 until s.length){
            var c = s[i]
            if(map.getOrDefault(c, -1) == -1){
                map[c] = i
                answer[i] = -1
            }else{
                answer[i] = i - map[c]!!
                map[c] = i
            }
        }
        println(map)
        return answer
    }
}

 

 

 

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

프로그래머스