[프로그래머스] 가장 큰 수 / C++

문제 설명

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;

bool compare(string a, string b)
{
    return (a + b) > (b + a);
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> nums;
    int i;
    
    for (i = 0; i < numbers.size(); i++)
    {
        nums.push_back(to_string(numbers[i]));
    }
    sort(nums.begin(), nums.end(), compare);
    if (nums.front() == "0")
    {
        return "0";
    }
    for (i = 0; i < numbers.size(); i++)
    {
        answer += nums[i];
    }
    return answer;
}

 

return값이 string 타입이기도 하고, 길이를 고려해 비교하기에 string 타입이 편하기도 해서 인자로 받은 int 벡터를 string 벡터로 변환했다.

처음엔 내림차순으로 정렬해서 차례대로 answer에 추가했었는데 두 번째 입출력 예를 통과하지 못했다. "3"과 "30"의 경우 "30"이 "0"으로 끝나므로 더 뒤에 있어야 한다. 이러한 경우를 처리하기 위해 compare 함수를 두 문자열을 더해 길이를 같게 한 다음 비교하는 방식으로 수정했다.

또한 모든 수가 "0"인 경우 결과는 "0"으로 이루어진 길이 n의 문자열이 아니라 "0"이어야 하므로 이 경우에 대한 예외를 처리해주었다.

 

링크

programmers.co.kr/learn/courses/30/lessons/42746

코딩테스트 연습 - 가장 큰 수

github.com/inbdni/Programmers/blob/master/level02/%EA%B0%80%EC%9E%A5%ED%81%B0%EC%88%98.cpp

inbdni/Programmers