[Programmers/Python] 가장 큰 수 || 정렬

프로그래머스

프로그래머스 파이썬 가장 큰 수 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 합니다.

📌 문제 풀이

1. numbers 리스트 안에 있는 원소들을 str로 변환한다.
2. fuctools.cmp_to_key를 이용하여 sorted에서의 정렬 기준을 재설정해준다.

• str인 원소를 붙여서 더 큰 수가 앞으로 정렬될 수 있도록 한다.

 

✅ [ CODE ]

import functools

def comparator(a, b):
    if int(a + b) > int(b + a):
        return -1
    else:
        return 1

def solution(numbers):
    numbers = list(map(str, numbers))
    numbers = sorted(numbers, key = functools.cmp_to_key(comparator))
    return str(int(''.join(numbers)))