특이한 정렬

2024. 8. 27. 18:00코딩테스트 입문

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


문제 설명
정수 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는 한 번에 한 개밖에 못 쓰기 때문에, 두 가지 이상의 규칙을 표현하고 싶다면 소괄호로 꼭 묶어주셔야 합니다.

'코딩테스트 입문' 카테고리의 다른 글

로그인 성공?  (0) 2024.08.29
등수 매기기  (2) 2024.08.28
유한소수 판별하기  (0) 2024.08.26
겹치는 선분의 길이  (0) 2024.08.25
평행  (0) 2024.08.24