[프로그래머스][C++] 가장 큰 수 (level2)

https://school.programmers.co.kr/learn/courses/30/lessons/42746?language=cpp

프로그래머스

 

문제

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"

 

※ 공지 - 2021년 10월 20일 테스트케이스가 추가되었습니다.

 

풀이

    answer = answer[0]=='0' ? "0" : answer;

위 처리를 해주지 않으면 (0으로 시작하는 문자열이 답인 경우) 테스트케이스 11에서 틀린다

 

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

using namespace std;

bool cmp(int a, int b) {
    string aa = to_string(a);
    string bb = to_string(b);
    
    string ab = aa + bb;
    string ba = bb + aa;
    
    return stoi(ab) > stoi(ba);
}

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