[프로그래머스] Lv.2 가장 큰 수 c++

https://school.programmers.co.kr/learn/courses/30/lessons/42746#

프로그래머스

 

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

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

 

풀이 방법

배열 안에 있는 모든 수를 이어붙여 가장 큰 수를 만들어야 합니다.

처음에 생각했던 방법은 DFS입니다. 하지만 [6, 10, 2]와 같은 입력의 경우, 6 10 2, 610, 606, 102, 210, 6102, 2106... 등 불필요한 경우까지 모두 계산하게 됩니다. 이렇게 되면 최대 원소 수가 10만인 이 문제에서는 비효율적이고 시간 초과가 발생할 수 있습니다.

 

그래서 두 번째로 생각한 방법은 정렬입니다.

그러나 단순히 큰 수 기준으로 정렬을 하면 "10"이 "6"보다 앞에 오게 되어, 원하는 결과를 얻을 수 없습니다.

이 문제를 해결하려면 정렬 함수 sort()를 커스텀해야 합니다. 먼저 vector<string> string_num을 선언하여 주어진 숫자들을 문자열로 변환한 뒤 이 벡터에 추가합니다. 그 후, sort() 함수를 사용해 정렬하되, 이때 비교 함수를 직접 정의합니다.

 

비교 함수는 bool cmp(string a, string b) 형태로 작성합니다. 두 문자열을 이어 붙였을 때 더 큰 값을 기준으로 정렬해야 하므로 return a + b > b + a를 사용합니다. 예를 들어, "610" > "106"이 참이므로 true를 반환합니다.

 

이런 식으로 정렬하면 최종적으로 ["6", "2", "10"]과 같이 문제의 목적에 맞게 정렬됩니다.

정렬된 결과를 순서대로 이어붙여 answer를 생성하면 되지만, 여기서 주의할 점은 중복된 숫자가 있는 경우입니다.

예를 들어, ["0", "0", "0"]과 같은 입력에서는 결과가 "000"이 될 수 있습니다. 따라서 이 경우는 "0"으로 변환해 반환해야 합니다.

 

풀이 코드

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

using namespace std;

vector<string> s_num;
bool cmp(string a, string b);
string solution(vector<int> numbers) {
    string answer = "";
    for(int i : numbers) s_num.push_back(to_string(i));
    sort(s_num.begin(), s_num.end(), cmp);
    for(string s : s_num) answer+=s;
    
    bool chk = true;
    for(char c : answer){
        if(c != '0') {
            chk = false;
            break;
        }
    }
    
    if(chk) answer="0";
    return answer;
}
bool cmp(string a, string b){
    return a+b > b+a;
}