chaerryos-blog

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

3 min read

문제

문제 설명

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”

풀이

오답

#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
string solution(vector<int> numbers) {
    string answer = "";
    // 문자열로 바꾸어서 정렬한다.
    vector<string> nums;
    for (int i=0; i<numbers.size(); i++) {
        nums.push_back(to_string(numbers[i]));
    }
    sort(nums.begin(), nums.end(), greater<string>());
    
    for (auto s:nums) {
        answer += s;
    }
    
    return answer;
}
테스트 1
입력값 〉[6, 10, 2]
기댓값 〉“6210”
실행 결과 〉테스트를 통과하였습니다.
테스트 2
입력값 〉[3, 30, 34, 5, 9]
기댓값 〉“9534330”
실행 결과 〉실행한 결괏값 “9534303”이 기댓값 “9534330”과 다릅니다.
문자열로 바꿔서 정렬하면, 자릿수가 적은 숫자가 뒤로 밀리게 된다.

정답 풀이

#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
string solution(vector<int> numbers) {
    string answer = "";
    // 문자열로 바꾸어서 정렬한다.
    vector<string> nums;
    for (int i=0; i<numbers.size(); i++) {
        nums.push_back(to_string(numbers[i]));
    }
    sort(nums.begin(), nums.end(), [] (string a, string b) {
        return a+b > b+a;
    });
    
    for (auto s:nums) {
        answer += s;
    }
    
    // 0밖에 없는 경우 "0"으로 바꿔서 출력해줘야 한다.
    if (answer[0] == '0') return "0";
    
    return answer;
}

Graph View