[프로그래머스/Kotlin] 가까운 수 (Lv. 0)

문제

정수 배열 array와 정수 n이 매개변수로 주어질 때, array에 들어있는 정수 중 n과 가장 가까운 수를 return 하도록 solution 함수를 완성해주세요.

 

제한사항

• 1 ≤ array의 길이 ≤ 100
• 1 ≤ array의 원소 ≤ 100
• 1 ≤ n ≤ 100
• 가장 가까운 수가 여러 개일 경우 더 작은 수를 return 합니다.

 

입출력 예

 

코드

import kotlin.math.*

class Solution {
    fun solution(array: IntArray, n: Int): Int {
        var answer: Int = 0

        array.sort()

        var minn: Int = 999999
        for(i in array){
            if(minn > abs(i-n)){
                minn = abs(i-n)
                answer = i
            }
        }
        println(minn)        

        return answer
    }
}

내 풀이는 위와 같이 일단 배열을 정렬한다.

그리고 배열만큼 반복하면서 해당 인덱스의 배열에서 n을 뺀 절댓값이 현재의 최소값보다 작으면 최소값으로 갱신한다.

다른 사람의 풀이를 보니 더 코틀린스러운(?) 코드가 있어서 가져왔다.

 

class Solution {
    fun solution(array: IntArray, n: Int): Int =
    array.sortedWith(compareBy<Int> { Math.abs(it - n) }
        .thenBy { it })
        .first()
}

sortedWith을 통해 정렬을 할건데 { } 에 있는 기준으로 정렬한다.

그리고 절대 차이가 같은 경우, thenBy { it }를 사용하여 값 자체에 따라 추가적으로 정렬합니다.

그리고 first를 사용해 첫번째 원소를 반환한다.

 

 

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

프로그래머스