[코딩테스트 입문]Day23-1
특이한 정렬
문제 설명
정수 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] |
문제 풀이
def solution(numlist, n):
new = {}
for j in numlist:
s = j - n
if s > 0:
new[str(j)] = abs(s)
elif s < 0:
new[str(j)] = abs(s) + 0.1
else:
new[str(j)] = abs(s)
vv = sorted(new.values())
convert_new = {v: k for k, v in new.items()}
return [int(convert_new.get(i)) for i in vv]
다른 풀이
def solution(numlist, n):
answer = sorted(numlist,key = lambda x : (abs(x-n), n-x))
return answer
def solution(numlist, n):
return sorted(numlist,key = lambda x: [abs(x-n),-x])
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]
댓글남기기