본문 바로가기
Algorithm/Sort

가장 큰 수

by HenryNoh 2021. 4. 28.

문제 설명

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"

구현

그리디 알고리즘처럼도 보이고 완전 탐색처럼도 보이는 이 문제의 핵심 요소는 제일 앞자리가 큰 수가 가장 먼저 와야 한다는 점이다.

예를 들어 [6, 10, 2]가 왔을 경우

  1. 세개의 수를 비교하여 앞자리가 가장 큰 6이 가장 먼저 오고
  2. 남은 두 개의 수를 비교하여 앞자리가 가장 큰 2가 먼저 오고
  3. 그다음 10이 와야 한다.

[3, 30, 34 5, 9]도 마찬가지이다.

  1. 5개 중에서 앞자리가 가장 큰 9가 먼저 온다.
  2. 4개 중에서 앞자리가 가장 큰 4가 먼저 온다.
  3. 3개 중에서 그다음 앞자리마저도 가장 큰 34가 먼저 온다.
  4. 남은 2개인 3과 30을 비교할 때는 330과 303을 비교해야 한다.

따라서 우리는 정렬을 앞에 오는 수 + 뒤에 오는 수 vs 뒤에 오는 수 + 앞에 오는 수로 비교를 하여야 한다.

그렇게 되면 자연스럽게 앞자리가 가장 큰 수가 앞으로 오게 되고 그다음 순서로 마찬가지로 될 것이다.


전체 코드는 아래에서 볼 수 있다.

#include <vector>
#include <string>
#include <algorithm>
using namespace std;

bool cmp(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));
    sort(temp.begin(), temp.end(), cmp);
    
    if (temp.at(0) == "0")        
        return "0";
    for (auto num : temp)
        answer += num;
    return answer;
}

 

댓글