수학, 알고리즘

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

춤추는수달 2022. 8. 4. 17:05

문제 설명

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

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

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

제한 사항
  • numbers의 길이는 1 이상 100,000 이하입니다.
  • numbers의 원소는 0 이상 1,000 이하입니다.
  • 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.
입출력 예numbersreturn
[6, 10, 2] "6210"
[3, 30, 34, 5, 9] "9534330"

※ 공지 - 2021년 10월 20일 테스트케이스가 추가되었습니다.

 


풀이

벡터에 담긴 수 들을 정렬하는 함수 따로 정의.

두 수를 비교할 때 합친 수가 더 큰 쪽의 앞에 온 수가 더 우선순위가 높음을 이용.

 

코드

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

using namespace std;

bool comp(int a, int b)
{
    string astr = to_string(a) ;
    string bstr = to_string(b) ;
    return stoi(astr+ bstr) > stoi(bstr+astr);
    
}

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