프로그래머스 / level.2 / 가장 큰 수(C++)

https://school.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"

 

내 해답

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

using namespace std;

string solution(vector<int> numbers) {
    string answer = "";
    int sum = 0;          
    vector<string> strings;
    
    // 문자로 변환
    for(int i = 0; i < numbers.size(); ++i)
    {
        strings.push_back(to_string(numbers[i]));
        sum += numbers[i];
    }
    
    // 0만 여러 개 있는 경우 예외처리
    if(sum == 0)
        return "0";
    
    // 두 문자를 더했을 때 더 큰 쪽을 우선하게 정렬 기준을 설정한다.
    // ex> 30 + 3 (303) < 3 + 30 (330)
    auto comp_ = [](auto a, auto b)
	{   
		return a + b > b + a;
	};
    
    // 정렬
    sort(strings.begin(), strings.end(), comp_);
    
    // 정답 문자열 만들기
    for(int i = 0 ; i < strings.size() ; ++i)
        answer += strings[i];
    
    return answer;
}