[프로그래머스] 가장 큰 수 / Python

문제

https://school.programmers.co.kr/learn/courses/30/lessons/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 합니다.

 

아이디어

 

처음엔 순열을 이용해 모든 가능한 수 를 만들고 max값을 출력했다.

정답을 구하는 측면에서 틀린 답이 아니지만, 메모리제한과 시간 제한이 있는 문제에 있어서 틀린 답이 되겠다. 

 

그래서 문자열 정렬을 이용해 내림차순으로 정렬해서 해결하려 했다. 숫자를 정렬하면 [30,9]가 되지만 문자열은 ASCII를 기준으로 정렬되기 때문에 []문제는 테케 2에 있는 [3,30,34,5,9]에서 나타났다.

 

그냥 정렬하게 되면 [9,5,34,30,3]이 되는데 9534303 < 9534330 이기 때문에 문제가 생긴다.

 

결론적으로 말하면 어떻게 해결할지 모르겠어서 구박사의 도움을 받았다..

각 number가 1,000이하이기 때문에 3자리로 맞추어 비교하겠다는 의미로 문자열에 *3을 해준다.

[333,303030, 343434, 555, 999]를 정렬하면 [9, 5, 34,3,30]이 된다. 

 

이렇게 결과를 출력하면 또 문제가 발생 하는데 문자열이기 때문에 [0,0,0,0]이 0이 아닌 '0000'으로 나타나기 때문에 생기는 문제다.

이런 경우를 없애기 위해 int로 형변환 했다가 다시 문자열로 바꾸는 작업을 해준다.

 

 

풀이

def solution(numbers):
    numbers = [srt(i) for i in numbers]
    numbers.sort(key=lambda x: x*3, reverse=True)
    return str(int(''.join(numbers)))