[프로그래머스] 특이한 정렬 (Python)

TK·2023년 10월 11일

문제 설명

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

제한사항

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

입출력 예

numlistnresult
[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]

입출력 예 설명

  • 입출력 예 #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이 앞에 와야 합니다.

정답 풀이1

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

정답 풀이2

def solution(numlist, n):
    answer = []
    for i in range(10000):
        if(n+i in numlist):
            answer.append(n+i)
        if(i!=0 and n-i in numlist):
            answer.append(n-i)
    return answer

정답 풀이3

def solution(numlist, n):
    # num -> (abs(n-num), -num)
    numlist = [(abs(n-num), -num) for num in numlist]
    # 첫 번째 요소를 기준으로 오름차순 정렬 and 두 번째 요소를 기준으로 내림차순 정렬
    numlist.sort()
    # 두 번쨰 요소만 반환
    return [-num for _, num in numlist]
profile
쉬운게 좋은 FE개발자😺

0개의 댓글