[Python3] 프로그래머스(Programmers) 가장 큰 수

문제 :

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

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

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

 

접근 :

이전에는 이 문제를 조합으로 풀어야 하는지 고민했었다. (당연히 시간 초과 나온다.)

시간 때문에 정렬로 풀려고 접근했고, 처음에는 numbers의 객체를 문자열로 바꾼 뒤에, arr.sort(key=lambda x: x, reverse=True)로 짜서 문제를 풀어봤다.

이렇게 되면, 3과 30을 비교할 때, 30이 더 큰 수로 비교하게 된다. 하지만 303보다, 330이 더 큰 수 이다.

이 문제를 해결하기 위해서 각자리에 3을 곱해준다. 그렇게 되면 3은 333(숫자가 아닌 문자열이기 때문에), 30은 303030으로 인식되어서, 3이 더 큰 수가 된다. 3을 곱해준 이유는 문제에서 제한은 1,000으로 두었기 때문이다.

 

그리고 정답을 제출했는데! 11번 테스트 케이스에서 틀렸다.

이유는 11번 테스트케이스가 [0, 0, 0, 0]이기 때문이다.

이 경우에는 '0000'이라고 정답이 입력된다.

 

따라서, str(int(ans))를 해줘서, '0000'을 0으로 만들고, 이를 다시 '0'으로 만들어야 한다. (문제에서 문자열로 제출하라고 함)

나는 그냥 0의 예외조건을 걸었다.

 

풀이 :

def solution(numbers):
    arr = []
    for i in numbers:
        a = str(i)
        arr.append(a)
    arr.sort(key=lambda x: x * 3, reverse=True)
    ans = ''
    for i in range(len(arr)):
        ans += arr[i]
    if int(ans) == 0:
        ans = '0'
    return ans # return str(int(''.join(arr)))