[Lv. 1] 가장 가까운 같은 글자

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/142086

프로그래머스

난이도 : Level.1

언어 : Kotlin

 

문제 설명 :

문자열 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은 영어 소문자로만 이루어져 있습니다.

입출력 예 :

s	result
"banana"	[-1, -1, -1, 2, 2, 2]
"foobar"	[-1, -1, 1, -1, -1, -1]

풀이 : 

class Solution {
    fun solution(s: String): IntArray {
        var answer: IntArray = intArrayOf()
        val charset = mutableSetOf<Char>()
        
        for (i in s.indices) {
            if (charset.contains(s[i])) {
                val indexArr = s.slice(s.indexOf(s[i]) .. i).mapIndexed { index, it -> if (it == s[i]) index else -1 }.filter { it != -1 }
                
                answer += indexArr[indexArr.lastIndex] - indexArr[indexArr.lastIndex - 1]
            } else {
                charset.add(s[i])
                answer += -1
            }
        }
        return answer
    }
}

 

내 풀이를 완전 간단하게 풀어낸 다른사람의 풀이도 가져와봤다.

 

다른 사람의 풀이 : 

class Solution {
    fun solution(s: String): List<Int> {
        return s.withIndex().map { (i, c) -> s.slice(0 until i).lastIndexOf(c).let { if (it >= 0) i - it else -1 } }
    }
}