[프로그래머스] 가장 큰 수(Python, Java)

https://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 합니다.

입출력 예

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

풀이 설명

1. numbers를 맨 앞자리가 큰 순서대로만 정렬하면 해결된다.

2. 문제의 첫 번째 예시로, 6과 10을 '666'과 '101010'으로 비교하고, 10과 2는 '101010'과 '222'로 비교하여 [6 2 10] 으로 정렬한다.

Python3 코드

def solution(numbers):
    numbers = list(map(str, numbers))
    numbers.sort(key=lambda x: x*3, reverse=True)
    
    if numbers[0] == "0":
        return "0"
    else:
        return ''.join(numbers)

Java 코드

import java.util.*;
class Solution {
    public String solution(int[] numbers) {
        String answer = "";
        int len = numbers.length;
        String[] arr = new String[len];
        for(int i=0;i<len;i++)
            arr[i] = Integer.toString(numbers[i]);
        Arrays.sort(arr,
                   (o1, o2) -> {return -Integer.compare(Integer.parseInt(o1+o2), Integer.parseInt(o2+o1));});
        if(arr[0].equals("0"))
            answer = "0";
        else
            for(String s: arr)
                answer += s;
        return answer;
    }
}