[프로그래머스] 정렬 Lv2. 가장 큰 수 (Python/파이썬)

[프로그래머스 코딩테스트 연습] 정렬 - 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 합니다.

입출력 예

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

제출

풀다가 잘 모르겠어서 검색해서 풀이를 찾아봤다. 보면서도 이해가 안가서 댓글 정독했다ㅎ

 

풀이 설명

1) 리스트에 numbers를 문자열로 변환하여 담음

2) lambda x: x * 3 을 해주는 이유는 제한 사항에 원소가 1000이하의 숫자이기 때문이다. 9와 991로 예를 들면, '9' * 3을 하면 '999'가 되는데 '991'과 비교했을 때 더 크다는 판단이 나온다. * 2를 해주면 '99' < '991' 이 되므로 * 3을 해줘야 원하는 정렬이 나올 수 있다. 

3) 리턴할 때, str(int())는 '0000'의 경우에는 '0'으로 반환해주고 문자열로 변환해야 하기 때문이다.

풀이 댓글 중 "str(int())가 필요없는 테스트케이스에도 적용이되서 시간복잡도가 높아지므로, 코드 맨위에 if sum(numbers) == 0: return '0' 만 작성해주시면 속도가 더 빨라집니다." 라는 의견도 있다. 

 

def solution(numbers):
    numbers = list(map(str, numbers))
    numbers.sort(key=lambda x: x * 3, reverse=True)
    # print(numbers)
    return str(int(''.join(numbers)))

어렵다 어려워,,,