[Lv.0/Python] 특이한 정렬

🔗 문제

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

코딩테스트 연습 - 특이한 정렬

 

문제 설명

정수 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]

 

💻 Code

def solution(numlist, n):
    return sorted(numlist, key=lambda x: (abs(n - x), -x))

sorted()를 이용하여 다음과 같은 기준에 따라 numlist를 정렬한다.

1. abs(n - x): n과의 거리가 작은 순으로 ➡️ 즉, |n - x|이 작은 순으로

2. -x: 만약 거리가 같다면, 더 큰 순으로 (내림차순으로 정렬하기 위해 -를 붙이는 것)