[정렬] 가장 큰 수

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

프로그래머스

---

문제 설명
0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.
예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.
0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

---

제한 사항

• numbers의 길이는 1 이상 100,000 이하입니다.
• numbers의 원소는 0 이상 1,000 이하입니다.
• 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

---

입출력 예

numbers	return
[6, 10, 2]	"6210"
[3, 30, 34, 5, 9]	"9534330"

---

숫자들을 이어붙여서 가장 큰 숫자를 만드는 문제입니다.

이런 문제들이 진짜 "모르면 못 푸는 문제"라고 생각합니다.

그래서 전 못 풀었습니다 ㅎㅎ...

 

functools 패키지에는 cmp_to_key라는 함수가 있습니다.

비교 함수를 키 함수로 바꿔주는 함수로, Python 2에서는 cmp 함수를 직접 사용할 수 있었지만, Python 3에서는 이를 키 함수로 변환해야 하기 때문에 사용해야 되는 함수죠.

우리가 새로 만든 compare이라는 함수를 기준으로 정렬할 때 사용합니다.

from functools import cmp_to_key

def compare(a, b):
    # ab가 ba보다 크다면
    if a + b > b + a:
        # 값을 바꾸지 않는다.
        return -1
    # ab가 ba보다 작다면
    elif a + b < b + a:
        # 값을 바꾼다
        return 1
    # 같으면 상관없다.
    else:
        return 0

def solution(numbers):
    # 모든 숫자를 문자열로 변환
    numbers = list(map(str, numbers))
    
    # 정렬
    numbers.sort(key=cmp_to_key(compare))
    
    # 정렬된 숫자들을 이어붙여 결과 생성
    answer = ''.join(numbers)
    
    # '0000'과 같은 경우를 방지하기 위해 int로 변환 후 다시 문자열로 변환
    return str(int(answer))

compare 함수를 바탕으로 numbers를 하나하나 비교해 나갑니다.

예를 들어 3과 30이 있다면, 330 > 303이므로 3이 30보다 앞에 나가게 됩니다.

이렇게 숫자를 비교해 나가면서 가장 큰 숫자를 만들어냅니다.

 

cmp_to_key 함수는 진짜 모르면 손도 못 대고 틀리는 문제라 생각이 드네요.

어떻게 정렬할까 고민했는데 참 어려웠습니다.