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

16 Oct 2018

Reading time ~2 minutes

문제

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”

풀이

정수 숫자들을 모두 string으로 바꿔 vector를 만들고, vector를 정렬한다.

이 때 정렬에 사용되는 compare 함수는 두 string a, b를 앞뒤로 붙였을 때 더 큰 순서로 정렬되도록 구현했다.

정렬 후 vector의 string들을 순서대로 이어 붙여 return한다.

#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> numstring;
    
    for (int i = 0; i < numbers.size(); i++) {
        numstring.push_back(to_string(numbers[i]));
    }
    
    sort(numstring.begin(), numstring.end(), compare);
    
    for (int i = 0; i < numstring.size(); i++) {
        answer += numstring[i];
    }
    
    if (answer[0] == '0')
        return "0";
        
    return answer;
}

출처: https://programmers.co.kr