[C++] 가장 큰 수

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"

<풀이>

이 문제의 핵심은 sort함수를 사용할 때, 3번째 인자로 비교하는 bool함수를 사용한다. 이 때, 문제의 핵심은 두 정수의 합을 비교할 때 더 큰 값이 앞으로 오도록 sort 할 수 있다. 즉, 6과 10이 있을 때, 두개를 더해서 610과 106을 비교해서 더 큰 것이 앞으로 나오도록 6, 10이 되도록 sort가 된다.

 

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

using namespace std;
bool compare(string a, string b) {
    return a+b>b+a;
}
string solution(vector<int> numbers) {
    string answer = "";
   
    vector <string> ans;
    for(int i=0;i<numbers.size();i++)
        ans.push_back(to_string(numbers[i]));
    
    sort(ans.begin(),ans.end(),compare);
    
    for(int i=0;i<ans.size();i++)
        answer+=ans[i];
    
    if(answer[0]=='0') answer="0";
    return answer;
}