https://programmers.co.kr/learn/courses/30/lessons/42746?language=python3
정렬 기본 문제
문제설명
0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.
예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.
0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.
입출력 예
numbers return
[6, 10, 2] 6210
[3, 30, 34, 5, 9] 9534330
솔루션
num은 0 이상 1,000 이하이다. (최대 4자리수)
num으로 만들 수 있는 가장 큰 4자리수는 (num으로*4)[:4]이다.
ex) 3->3333 34->3434 30->3030 ====> 34330
이를 기준으로 정렬하면 이어 붙였을때 가장 큰 값을 얻을 수 있다.
코드
# 파이썬
def solution(numbers):
numbers = list(map(str, numbers))
numbers.sort(key= lambda x : (x*4)[:4],reverse=True)
return '0' if numbers[0]=='0' else ''.join(numbers)
주의
모든 숫자가 0 인경우 (numbers[0] == 0) join을 해주면 안되고 0을 return 해야한다.