Programmers <가장 큰 수>


Level 2

문제 설명

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"

작성 코드

#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> temp;
    
    for (int i = 0; i < numbers.size(); i++) {
		temp.push_back(to_string(numbers[i]));
	}

	sort(temp.begin(), temp.end(), compare);

	for (int i = 0; i < temp.size(); i++)
		answer += temp[i];

	if (temp[0] == "0")
		answer = "0";
    
    return answer;
}

📌

  • 사용자 정의 함수를 이용한 sort()를 활용하는 문제이다.
  • vector temp를 정렬할 때, 더한 문자열이 큰 순서로 정렬하도록 하는 사용자 정의 함수가 compare()이다.
  • sort 함수는 기본적으로 오름차순으로 요소들을 정렬하지만, 사용자 정의 함수를 통하여 정렬 순서를 바꿀 수 있다.
  • 처음 문제를 풀 때에는 모든 경우의 수를 고려하여 확인해야 하나 하고 생각했지만 사용자 정의 함수 sort를 통해 쉽게 정렬한 후 최종 문자열을 얻어낼 수 있었다.
  • next_permutation을 통하여 주어진 문자열에 대한 순열을 가지고 가장 큰 수를 max 변수의 update를 통해 알아내는 방법도 있을 것 같다.
  • 주의할 점은 가장 앞자리가 0인 경우를 고려해야한다는 것!

좋은 웹페이지 즐겨찾기