[Algorithm] 프로그래머스 - 특이한 정렬 (Kotlin)

정수 n을 기준으로 n과 가까운 수부터 정렬하려고 한다. 이 때 n으로부터의 거리가 같으면 더 큰 수를 앞에 오게 배치한다. 정수가 담긴 배열 numlist, 정수 n이 주어질 때 numlist의 원소를 n으로부터 가까운 순서대로 정렬한 배열을 리턴하는 solution()을 완성하라

 

 

이 문제도 다른 사람의 풀이를 참고했다.

 

https://gyulpiki.tistory.com/entry/%ED%8A%B9%EC%9D%B4%ED%95%9C-%EC%A0%95%EB%A0%AC-Java-%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%EC%9E%85%EB%AC%B8

특이한 정렬 - Java [프로그래머스 입문]

import kotlin.math.abs

class Solution {
    fun solution(numlist: IntArray, n: Int): IntArray {
        val arr = ArrayList<Int>()
        for (element in numlist) {
            arr.add(Integer.valueOf(element))
        }

        arr.sortWith { a, b ->
            val absA = abs(a - n)
            val absB = abs(b - n)
            if (absA == absB) {
                if (a > b) {
                    -1
                } else {
                    1
                }
            } else {
                absA - absB
            }
        }

        val answer = IntArray(numlist.size)
        for (i in arr.indices) {
            answer[i] = arr[i]
        }
        return answer
    }
}

 

간단하게 만들면 아래처럼 된다.

 

import kotlin.math.abs

class Solution {
    fun solution(numList: IntArray, n: Int): IntArray {
        return numList.sortedWith { a, b ->
            if (abs(a - n) == abs(b - n)) b.compareTo(a) else abs(a - n).compareTo(abs(b - n))
        }.toIntArray()
    }
}