[정렬] 가장 큰 수

문제)

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

 

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"

 

 

 

 

접근)

 sort 할 때 사용자 정의 함수를 사용하는 것이 관건이었다. s1+s2 > s2+s1으로 정의해주면 두개의 문자를 더했을 때 더 큰 순서대로 정렬하게 된다. 예를 들어 "3"과 "12"가 있을 때 "3" + "12" = "312" > "12" + "3" = "123" 이므로 "3", "12" 순서가 된다. 이렇게 정렬이 끝나면 이 순서대로 answer에 붙이면 된다.

 여기서 주의할 점은 만약 numbers가 [0, 0, 0]인 경우이다. answer이 "000"으로 나오게 되는데 이 예외처리를 해주기 위해서는 정렬이 다 끝났을 때 가장 처음 수가 "0"인 경우는 바로 "0으로 답을 리턴하면 된다.

 

 

 

풀이)

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

bool cmp(string s1, string s2){
    return s1+s2>s2+s1; //오름차순
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> v;
    for(int i: numbers){
        v.push_back(to_string(i));
    }
    sort(v.begin(), v.end(), cmp);
    if(v[0]=="0") return "0";
    for(string s: v){
        answer+=s;
    }
    return answer;
}