[Lv. 0] 특이한 정렬

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

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

• 정수 n을 기준으로 n과 가까운 수부터 정렬하려고 합니다. 이때 n으로부터의 거리가 같다면 더 큰 수를 앞에 오도록 배치합니다. 정수가 담긴 배열 numlist와 정수 n이 주어질 때 numlist의 원소를 n으로부터 가까운 순서대로 정렬한 배열을 return하도록 solution 함수를 완성해주세요.

제한 사항 : 

• 1 ≤ n ≤ 10,000
• 1 ≤ numlist의 원소 ≤ 10,000
• 1 ≤ numlist의 길이 ≤ 100
• numlist는 중복된 원소를 갖지 않습니다.

입출력 예 :

numlist	n	result
[1, 2, 3, 4, 5, 6]	4	[4, 5, 3, 6, 2, 1]
[10000, 20, 36, 47, 40, 6, 10, 7000]	30	[36, 40, 20, 47, 10, 6, 7000, 10000]

풀이 : 

import kotlin.math.abs

class Solution {
    fun solution(numlist: IntArray, n: Int): IntArray = numlist.sortedWith(compareByDescending { abs(it - n) }).reversed()
}

처음 풀이는 다음과 같다.

이전에 풀었던 '가까운 수'라는 문제가 떠올라 쉽게 처리했으나, 테스트 케이스 2번을 실패하게 되었다.

왜? 라는 생각에 질문글을 통해 테스트 2 반례를 구하고 나니 알 수 있게 되었다.

 

반례 : 

n = 500
lsit = [600, 400, 300, 200, 700, 800, 100, 900]
기대값 = [600, 400, 700, 300, 800, 200, 900, 100]

 

근데 나는 출력해보니[400, 600, 700, 300, 800, 200, 900, 100]으로 나오는 것이었다.

코드 수정이 필요했다.

 

 

1. 내림차순 제거

numlist.sortedWith(compareBy<Int> { abs(it - n) })

결과값 = [600, 400, 300, 700, 200, 800, 100, 900]>

입출력 예 1번 케이스 결과값 = [4, 3, 5, 2, 6, 1]

 

식별한 문제점은 '내림차순과 비교를 동시에 하면 안되는가'였다.

따라서 비교를 먼저 한 뒤, 내림차순을 시행하면 문제 없이 실행 되리라 생각했다.

import kotlin.math.abs

class Solution {
    fun solution(numlist: IntArray, n: Int): IntArray = numlist.sortedWith(compareBy<Int> { abs(it - n) }.thenByDescending { it }).toIntArray()
}

 

참고 레퍼런스 : 

https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.comparisons/compare-by-descending.html

compareByDescending - Kotlin Programming Language

https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.comparisons/then-by-descending.html

thenByDescending - Kotlin Programming Language