문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.
예를 들어, s="banana"라고 할 때, 각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.
문자열 s이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.
fun solution(s: String): IntArray {
var answer = IntArray(s.length) { -1 }
var arr = s.toCharArray()
var lastIndex = mutableMapOf<Char, Int>()
for (i in arr.indices) {
if (lastIndex.containsKey(arr[i])) {
answer[i] = i - lastIndex[arr[i]]!!
}
lastIndex[arr[i]] = i
}
return answer
}
answer를 -1 로 초기화 해주고 s를 Char배열로 받아온다
문자와 위치를 저장할 맵 을 만들어주고
containsKey()
를 써 이미 나온 문자가 있는지 확인하고 true라면 현재 i에서 위치값을 빼주고 answer에 넣어준다
fun solution(s: String): List<Int> {
return s.withIndex().map{(i,w) ->
s.slice(0 until i).lastIndexOf(w).let{
if(it >= 0) i-it else it }
}
}
withIndex
로 각요소의 인덱스를 위치값으로 정해주고
문자열을 0부터 i미만까지 슬라이스 하고 w문자가 나온 마지막 위치를 반환하는데
그값(it)이 0보다 크다면 이전에 같은 문자가 나타났다는 뜻이고 i에서 빼준다
나온적이없다면 -1을 반환한다