Python: sort, sorted 활용하기

정렬 기준 선택

오름 차순 정렬

nums = [[3, 4], [1, 1], [1, -1], [2, 2], [3, 3]]
sort(nums, key=lambda x: (x[0], x[1]))

 정렬하고자 하는 기준이 여러 개인 경우 위와 같이 우선적으로 정렬할 기준을 선택하여 정렬할 수 있다. 반대로 `x: (x[1], x[0])`가 된다면 2번째 값을 기준으로 정렬을 하게 된다.

 

내림 차순 정렬

nums = [[3, 4], [1, 1], [1, -1], [2, 2], [3, 3]]
sort(nums, key=lambda x: (-x[0], x[1]))

 정렬하고자 하는 기준에 하나는 내림차순, 하나는 오름 차순으로 정렬하고자 한다면 위와 같이 `-`를 붙여주면 내림차순으로 정렬할 수 있다.

 

중복되는 값이 있는 경우

from collections import defaultdict

nums = defaultdict(int)
n = int(input())

for i in range(n):
    num = int(input())
    nums[num] += 1

for num in sorted(nums):
    print((str(num) + ' ') * nums[num], end='')

 n이 증가할 수록 정렬의 오버헤드가 증가하므로 해당 수가 몇 개인지 카운트하여 정렬 후 해당 숫자만큼 출력하게 되면 보다 효율적으로 정렬을 진행할 수 있다.