[Programmers] 가장 큰 수

문제 설명

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

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

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

제한 사항

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

소스코드

C++

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

// 순서 배치를 위한 함수
bool cmp(string a, string b) {
    return (a+b) > (b+a);
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> tempNum;
    for(auto num : numbers) {
        tempNum.push_back(to_string(num));
    }
    sort(tempNum.begin(), tempNum.end(), cmp);
    for(auto tmp : tempNum) {
        answer += tmp;
    }
    // 0인 경우
    if(answer[0] == '0') {
        answer = '0';
    }
    return answer;
}

Java

import java.util.*;
class Solution {
    public String solution(int[] numbers) {
        String answer = "";
        String[] numstr = new String[numbers.length];
        
        // int -> String
        for(int i = 0; i < numbers.length; i++) {
            numstr[i] = Integer.toString(numbers[i]);
        }
        
        Arrays.sort(numstr, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                return (s2+s1).compareTo(s1+s2); // 내림차순
            }
        });
        
        if("0".equals(numstr[0])) return "0";
        
        for(String num : numstr) {
            answer += num;
        }
        return answer;
    }
}

소감

C++ sort 함수의 사용자 정의 함수를 통한 정렬에 대해 다시 한 번 학습할 수 있었던 문제였다.