프로그래밍 공부
작성일
2024. 1. 23. 12:16
작성자
WDmil
728x90

문제 설명

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"

첫 시도

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

bool compare(const std::string &a, const std::string &b) {
    return a + b > b + a;
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> sNumbers;
    
    for(auto& def : numbers) 
        sNumbers.push_back(to_string(def));
    
    sort(sNumbers.begin(), sNumbers.end(), compare);
    
    for(const auto& def : sNumbers)
        answer += def;

    return answer;
}

실패

 

알고리즘 자체는 문제가 없으나, 문제 자체가 악의적이었다.

문항의 숫자가 [0, 0, 0]이 들어갈 경우 "0"이 출력되어야 하나 "000"이 출력되면 실패


두번째 시도

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

bool compare(const std::string &a, const std::string &b) {
    return a + b > b + a;
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> sNumbers;
    
    for(auto& def : numbers) 
        sNumbers.push_back(to_string(def));
    
    sort(sNumbers.begin(), sNumbers.end(), compare);
    
    for(const auto& def : sNumbers)
        answer += def;

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

성공

 

그냥 첫 문자가0일경우 0으로 바꿔버리는 if문을 정의함.

728x90

'코딩테스트 문제 풀이 > 정렬' 카테고리의 다른 글

테이블 해시 함수  (0) 2024.05.15
H-Index  (0) 2024.02.01
K번째수  (0) 2024.01.12