[C++] 프로그래머스 - "가장 큰 수" 풀이

문제 설명

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

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

 

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

 

 

문제 주소: https://programmers.co.kr/learn/courses/30/lessons/42746

 

문제 풀이

numbers의 원소들을 string 타입으로 바꿔서 정렬을 실행하면 앞자리가 큰 순서대로 정렬된다. 

하지만 {"3", "30"}의 경우 {"3", "30"}으로 정렬되어야 하는데 {"30", "3"}으로 정렬되므로 이를 처리해줘야 한다. 

비교함수 comp를 만들어서 두 string을 더했을때(연결했을 때) 값이 커지는 경우 true를 반환하도록 해주면 된다.

 

답안을 제출했을 때 테스트11을 통과하지 못했는데, {0, 0}과 같은 경우 예외처리를 안해줬기 때문이었다. 

numbers가 {0, 0}일 경우 answer이 "0"이 되어야하는데 "00"이 되므로 이를 처리해줘야 한다.

따라서 temp의 첫번째 원소가 "0"으로 시작하면 "0"을 return하도록 코드를 추가했다. 

 

 

소스 코드

#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
bool comp(string a, string b) {
    return a + b > b + a;
}
 
string solution(vector<int> numbers) {
    string answer = "";
    vector<string> temp;
 
    for (auto num : numbers)
        temp.push_back(to_string(num)); // numbers 원소를 string형으로 바꿔서 저장
    sort(temp.begin(), temp.end(), comp);
 
    if (temp[0] == "0")
        return "0";
    else
        for (const auto num : temp)
            answer += num;
 
    return answer;
}