Study/Coding Test

[정렬] 가장 큰 수

_gayeon 2021. 2. 2. 17:03

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

 

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

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요. 예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰

programmers.co.kr

문제 설명

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

Sol) 숫자를 문자로 변환하기

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

using namespace std;

bool comp(string a, string b){
    if(a+b > b+a)
        return true; 
    else
        return false;
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> v;
    
    for(int i=0; i<numbers.size(); i++){
        v.push_back(to_string(numbers[i]));
    }
    
    sort(v.begin(), v.end(), comp);
    for(int i=0; i<v.size(); i++){
        answer+=v[i];
    }
    
    if (answer[0] == '0')  // 0만 여러 개인 경우
		return "0";
    
    return answer;
}

문제의 정렬 방식은 문자열 정렬방식과 동일하다.

단, 3과 30 중 30이 더 크다고 정렬이 되므로 비교함수를 따로 작성해야 한다. 

 

마지막, 0만 여러 개인 경우도 예외처리를 꼭 해줄것.