코딩 테스트

[프로그래머스 / C++] 가장 큰 수

haram.jeon 2022. 2. 10. 23:02

https://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"

 

문제 해결

3, 30 이 있을때 330, 303 둘중 어느것이 큰지 확인할 수 있는것이 포인트. 단순 정렬하면 30, 3 이 되기 때문에 어느것이 앞에 왔을때 더 큰지 확인하게끔 한다.

나는 sum을 다 더해서 0인 경우에 0을 return 하도록 했는데 굳이 더 할 필요없이 answer가 0으로 시작하는지 보면 불필요한 연산이 제거되겠다.

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

using namespace std;

string solution(vector<int> numbers) {
	string answer = "";

	auto comp_ = [](auto a, auto b)
	{
		auto str_a = to_string(a);
		auto str_b = to_string(b);
        
		return str_a + str_b > str_b + str_a;
	};

	sort(numbers.begin(), numbers.end(), comp_);

	size_t sum = 0u;
	for (auto& num : numbers)
	{
		answer += to_string(num);
		sum += num;
	}

	if (sum == 0u)
		answer = "0";

	return answer;
}

결과