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

코딩테스트 연습 - 가장 큰 수

문제 설명

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

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

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

제한 사항

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

입출력 예시

풀이 과정

생각보다 쉽게 풀 수 없는 문제였다. 처음에는 자리수를 최대로 같게 만들어 검사하면 되겠다고 생각했지만, 예를들어 3이면 3000으로 만들었다. 하지만 이렇게 푸는 방법은 정렬이 올바르게 되지 않았다. 그래서 1000이하의 수들을 4자리 수로 만들어 예를 들어 3이면 3333, 121 이면 1211로 tmp_number에 저장한 다음 answer에 str(tmp_number)를 저장하였다.

그다음 answer를 sort 해준뒤 str로 하나씩 이어붙여주었다.

이때 [0,0,0,0] 일때는 0이 출력되어야 하므로 int(ans)가 0이면 str(0)을 출력하도록 했다.

def solution(numbers):
    answer = []
    ans = ''
    for number in numbers:
        tmp_number = [str(i) for i in str(number)]
        while len(tmp_number) < 4:
            for i in str(number):
                tmp_number.append(i)
            tmp_number = tmp_number[0:4]
        answer.append([str(''.join(tmp_number)), number])
    answer.sort(reverse=True)
    for _, item in answer:
        ans += str(item)
    return str(0) if not int(ans) else ans

if __name__ == '__main__':
    print(solution([0,0,0,1000]))
    print(solution([6,10,2]))
    print(solution([0,0,0,0]))