2024. 8. 27. 18:00ㆍ코딩테스트 입문
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] |
입출력 예 설명
입출력 예 #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이 앞에 와야 합니다.
정렬은 필요한데, 단순히 오름차순/내림차순이 아니라 특정 n에서 떨어진 거리를 기준으로 정렬을 해야 하는 문제입니다.
이런 특정 규칙에 따른 정렬 문제는 정렬에 필요한 key를 따로 설정하는 작업이 필요한데요, 이 문제는 그나마 간단하게 조건이 2가지 뿐입니다.
1. n에서 떨어진 거리를 기준으로 정렬
2. 같은 거리일 경우, 더 큰 값이 우선 정렬
정렬에 key를 적용하는 함수는 sorted(list, key) 구조입니다.
key를 설정해서 문제를 풀어보도록 하겠습니다. 파이썬을 사용합니다.
def solution(numlist, n):
# 1번 정렬: n에서 떨어진 절댓값 거리
# 2번 정렬: 큰 값을 우선 정렬
return sorted(numlist, key=lambda x:(abs(x - n), -x))
numlist를 정렬하는데, key로 두 가지 규칙이 적용됩니다.
규칙은 왼쪽에서 오른쪽으로 진행되며, 왼쪽에서 같은 값들은 오른쪽 정렬이 적용됩니다.
첫 번째 규칙은 abs(x - n)입니다. n에서 떨어진 거리(절댓값)를 기준으로 오름차순 정렬합니다.
두 번째 규칙은 -x입니다. -x는 내림차순 정렬을 의미합니다.
즉, n에서 떨어진 거리를 기준으로 오름차순 정렬을 하고, 같은 거리라면 내림차순으로 정렬한다는 규칙을 나타냅니다.
참고로 key는 한 번에 한 개밖에 못 쓰기 때문에, 두 가지 이상의 규칙을 표현하고 싶다면 소괄호로 꼭 묶어주셔야 합니다.