[프로그래머스] Lv2. 가장 큰 수

문제 설명

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

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

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

 

제한 사항

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

풀이

파이썬의 문자열 정렬은 사전순으로 일어난다는 것을 핵심으로 문제 풀이를 진행하였습니다.

숫자의 범위는 0~1000까지의 정수이기에 1의 자리를 갖는 수는 최대 자기자신을 3번 이어붙이면 범위 내의 모든 숫자와 사전 비교 정렬을 할 수 있기에 lambda를 사용하여 진행하였으며, 최종적으로 정렬된 리스트를 join을 통해 이어붙여 완성하였습니다.

빠트리지 말아야 할 사항으로는 맨 앞 숫자가 0일 경우, 0이 답이라는 경우를 꼭! 놓치지 말아야합니다!!!

def solution(numbers):
    numbers = list(map(str, numbers))
    numbers.sort(key=lambda x: x*3, reverse=True)
    answer = ''.join(numbers)
    return '0' if answer[0] == '0' else answer