[Algorithm] 프로그래머스 - 숫자 찾기 (Kotlin)

정수 num, k가 매개변수로 주어질 때, num을 이루는 숫자 중 k가 있으면 num의 그 숫자가 있는 자리수를 리턴하고 없으면 -1을 리턴하는 solution()을 완성하라

 

 

입출력 예를 잘 봐야 한다. 29183에서 1은 2번째가 아닌 3번째에 있다. 즉 num에 k가 있으면 k의 인덱스에 +1을 해야 한다는 뜻이다.

아래는 주먹구구식으로 푼 풀이다.

 

class Solution {
    fun solution(num: Int, k: Int): Int {
        val nums = num.toString()
        val ks = k.toString()
        var result = 0
        if (nums.contains(ks)) {
            result += nums.indexOf(ks) + 1
        } else {
            result = -1
        }

        return result
    }
}

 

이것을 줄여보려고 했는데 잘 안 되서 다른 사람의 풀이를 참고했다.

 

class Solution {
    fun solution(num: Int, k: Int) = "-$num".indexOf(k.toString())
}