본문 바로가기

알고리즘/프로그래머스

[ Python ] - 특이한 정렬(120880)

728x90
반응형

문제 설명

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


제한사항

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

입출력 예


입출력 예 설명

입출력 예 #1 

  • 4에서 가까운 순으로 [4, 5, 3, 6, 2, 1]을 return합니다. 
  • 3과 5는 거리가 같으므로 더 큰 5가 앞에 와야 합니다. 
  • 2와 6은 거리가 같으므로 더 큰 6이 앞에 와야 합니다. 

입출력 예 #2 

  • 30에서 가까운 순으로 [36, 40, 20, 47, 10, 6, 7000, 10000]을 return합니다. 
  • 20과 40은 거리가 같으므로 더 큰 40이 앞에 와야 합니다.

solution.py

def solution(numlist, n):
    # 각 숫자와 n과의 절대 차이를 튜플로 만들어 리스트에 저장한다.
    # 이 때 튜플의 첫 번째 요소는 원래 숫자, 두 번째 요소는 절대 차이 값이다.
    sorted_list = sorted([(num, abs(num - n)) for num in numlist], 
                         # 정렬 기준은 먼저 절대 차이 값(x[1])이 작은 순,
                         # 그 후 같은 차이 값을 가진 경우는 숫자(x[0])가 큰 순으로 정렬한다.
                         key=lambda x: (x[1], -x[0]))
    
    # 정렬된 리스트에서 원래 숫자들만 추출하여 반환한다.
    return [num[0] for num in sorted_list]
728x90
반응형