[프로그래머스] 가장 큰 수 STL(sort)

문제 :

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

코딩테스트 연습 - 가장 큰 수

 

분석 : 

우선 가장 첫번째 자리가 큰 원소 먼저 정렬 되어야 하는 것으로 보인다 !

하지만 문제는 첫번째 자리 원소가 같은 원소들이다.  

예를 들어 [8, 3, 30, 34, 36] 가 있을때 기본 sort함수로 정렬하면 [8, 36, 32, 30, 3] 순으로 정렬된다. 

하지만 32303보다 33230이 더 큰 수 이며 이를 구별해줄 방법이 필요했다.

comp 함수에 length가 더 크고, while문 써서 index 늘리고 ... 하다보니 코드가 지엽적으로 변해서 문제였다. 그러다가 완전 명쾌한 해답을 발견했다 ! 참고 : https://mungto.tistory.com/22 

문제에서 주어진 대로 비교하는 문자열과 문자열끼리 + 연산을 한뒤 큰 값을 리턴하게 한다. 완전 굿 방법 ..!

 

 c++ 코드 :

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


using namespace std;
bool comp(string s1, string s2){
	return s1 + s2 > s2 + s1;
}

string solution(vector<int> numbers) {
    string answer = "";
    vector<string> num;
    
    for(int i = 0; i < numbers.size(); i++){
        num.push_back(to_string(numbers[i]));
    }
    sort(num.begin(), num.end(), comp);
    if (num.at(0) == "0") {
        return "0";
    }
    for(int i = 0; i < numbers.size(); i++){
        answer += num[i];
    }
    
    return answer;
}

 

결과 :