(C++) 프로그래머스 level2 : 가장 큰 수

 

문제 설명:

 

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

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

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

 

제한 사항

 

numbers의 길이는 1 이상 100,000 이하입니다.

numbers의 원소는 0 이상 1,000 이하입니다.

정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

 

입출력 예

• 0열 선택0열 다음에 열 추가
• 1열 선택1열 다음에 열 추가

• 0행 선택0행 다음에 행 추가
• 1행 선택1행 다음에 행 추가
• 2행 선택2행 다음에 행 추가

셀 전체 선택

열 너비 조절

행 높이 조절

numbers	return
[6, 10, 2]	"6210"
[3, 30, 34, 5, 9]	"9534330"

출처: https://programmers.co.kr/learn/courses/30/lessons/42746

 

 

---

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

bool compare(string a, string b)
{
	return a + b > b + a;
}

string solution(vector<int> numbers) {

	vector<string>nums;
	for (int i = 0; i < numbers.size(); i++)
		nums.push_back(to_string(numbers[i])); //각 원소를 문자열로 변환
	sort(nums.begin(), nums.end(), compare); //compare로 정렬
	string answer = "";
	for (int i = 0; i < nums.size();i++)
		answer += nums[i]; //문자열 합치기
	if (answer[0] == '0') return "0"; //답이 0인 경우
	return answer;
}

 

 

답이 0인 경우를 생각하지 못했어서 처음에 오류가 났었다.

 

이러한 경우를 처리하고 나니 정답!

 

문자열로 비교할 때는 첫번째 문자부터 차례로 검사하므로 a+b > b+a 식을 이용하여 간단하게 풀 수 있었다.

 

내가 생각해낸 아이디어였는데, 프로그래머스 대표정답으로 올라온 답과 똑같아서 기분이 좋았다 ㅎ