[프로그래머스] 가장 큰 수 - python

가장 큰 수

문제 설명

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.

예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

제한사항

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

풀이

numberd의 길이가 100,000이기 때문에 하나씩 조합하면서 비교할 수 없다. 가장 큰 수를 만들기 위해서는 각 첫자리가 더 큰 수가 앞에 오도록 만들어야한다.

numbers의 원소가 1000로 자릿수가 크기 때문에 radix sort는 사용할 수 없고 문자열 비교 방식으로 해결할 수 있다.

최대 1000이므로 자릿수를 맞추기 위해 각 숫자를 3개씩 이어 붙여서 비교한다.

만약 모든 원소가 0일 경우 "0000"으로 출력될 수 있기 때문에 예외처리를 해준다.

• Int -> str 로 변환하는 방법도 있지만 이 변환도 비용이 클 수 있기 때문에 먼저 예외처리를 해주면 더 빠르다

def solution(numbers):
    if sum(numbers) == 0 :
        return "0"
    answer = sorted([str(n) for n in numbers], key=lambda x: x * 3, reverse=True)
    return "".join(answer)