와챠의 우당탕탕 코딩 일기장
[kotlin] 프로그래머스 - 특이한 정렬 / sortedWith 본문
반응형
문제 설명
정수 n을 기준으로 n과 가까운 수부터 정렬하려고 합니다. 이때 n으로부터의 거리가 같다면 더 큰 수를 앞에 오도록 배치합니다. 정수가 담긴 배열 numlist와 정수 n이 주어질 때 numlist의 원소를 n으로부터 가까운 순서대로 정렬한 배열을 return하도록 solution 함수를 완성해주세요.
입출력 예
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] |
풀이
sortedWith 함수를 사용해 요소의 정렬 기준을 n과 가까운 것으로 하도록 설정했다.
import kotlin.math.abs
class Solution {
fun solution(numlist: IntArray, n: Int): IntArray = numlist.sortedWith { a, b ->
// n으로부터의 거리가 같다면 더 큰 수를 앞에 오도록 배치
if (abs(a - n) == abs(b - n)) b.compareTo(a)
// n으로부터 가까운 순서대로 정렬
else abs(a - n).compareTo(abs(b - n))
}.toIntArray()
}
sortedWith
- sortedBy와 마찬가지로 원본 배열을 수정하지 않고 정렬한 배열을 리턴한다.
- sortedBy와 다르게 정렬 기준을 여러 개 둘 수 있다.
- first 기준 정렬 후 second 기준 정렬
list.sortedWith(compareBy({ it.first }, { it.second }))
- 정렬
- 아래 두 방법 모두 오름차순으로 정렬하는 방법이다.
- 내림차순은 b.compareTo(a)로 바꾸면 된다.
// [10000, 20, 36, 47, 40, 6, 10, 7000]
// -> [6, 10, 20, 36, 40, 47, 7000, 10000]
numlist.sortedWith { a, b -> if(a > b) 1 else -1 }
numlist.sortedWith { a, b -> a.compareTo(b) }
- compareTo
- a, b 차이에 따라 0, 1, -1 중 하나를 반환한다.
- a == b: 0
- a > b: 1
- a < b: -1
반응형
'코딩 일기장 > Android(Kotlin)' 카테고리의 다른 글
[Compose] 2-1. Material Design 3으로 앱 테마 지정 / Material Theme Builder / Color, Type, Theme, Shape 설정 (0) | 2025.04.16 |
---|---|
[kotlin] 프로그래머스 / 택배 상자 꺼내기 (0) | 2025.04.16 |
[Compose] 1-3. Compose State/Stateful/Stateless/ViewModel/remember* (0) | 2025.04.04 |
[kotlin] 프로그래머스 - 안전지대 (1) | 2025.04.03 |
[Compose] 1-2. Compose의 기본 레이아웃/element/list/NavigationBar/window size (0) | 2025.04.01 |
Comments