프로그래머스 (가장 큰 수, 정렬) C++

 

일단 순열로 접근을 했는데 몇 개의 Test case가 돌아가지 않아서 좀 답답했다. Sort를 할 때 비교 우선순위를 부여하니까 바로 풀렸던 문제

 

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

using namespace std;

bool cmp(const string& a, const string& b) {
    if(a + b > b + a) return true;
    else return false;
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> real;
    
    // 벡터에 string으로 할당
    for(int i = 0 ; i < numbers.size() ; i++) {
        real.push_back(to_string(numbers[i]));
    }
    
    // 오름차순으로 숫자를 정렬
    sort(real.begin(), real.end(), cmp);
    
    // answer 문자열을 만들어줌
    for(int i = 0 ; i < real.size() ; i++) {
        answer += real[i];
    }
    
    if(answer[0] == '0')
        return "0";
    else
        return answer;
}

 

 

next_permutation을 사용해서 풀어봤던 문제

 

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

using namespace std;

string solution(vector<int> numbers) {
    string answer = "";
    string real = "";
    int max = 0;
    
    // 오름차순으로 숫자를 정렬
    sort(numbers.begin(), numbers.end());
    
    // 순열조합을 이용
    do {
        // 각 숫자를 문자열로 변환해서 answer에 담아주기
        for(int i = 0 ; i < numbers.size() ; i++) 
            answer += to_string(numbers[i]);
        
        // answer가 최댓값인 경우 계속 최신화시켜준다.
        if(max < stoi(answer))
            max = stoi(answer);
        
        // answer 초기화
        answer = "";
    } while(next_permutation(numbers.begin(), numbers.end()));
    
    real = to_string(max);
    
    if(real[0] == '0') return "0";
    else
        return real;
}