백준 1431 _ 시리얼 번호

https://www.acmicpc.net/problem/1431

1431번: 시리얼 번호

---

⭐ 내 소스 코드

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

//정렬을 위한 사용자 정의 함수 지정
bool compare(string serial1, string serial2) {
    int sum1 = 0, sum2=0;

    //두 시리얼의 길이가 같지 않을 경우
    if (serial1.length() != serial2.length()) {
        //길이 순으로 정렬
        return serial1.length() < serial2.length();
    }

    //두 시리얼의 길이가 같을 경우
    else if(serial1.length() == serial2.length()){
        for (int i = 0; i < serial1.length(); i++) {
            //시리얼 문자들 중 숫자의 값만 계산
            if (isdigit(serial1[i])) {
                sum1 += serial1[i] - 48;
            }
        }
        for (int i = 0; i < serial2.length(); i++) {
            if (isdigit(serial2[i])) {
                sum2 += serial2[i] - 48;
            }
        }
        //두 시리얼 번호의 숫자 합계가 같지 않을 경우
        if (sum1 != sum2) {
            //합계 순으로 정렬
            return sum1 < sum2;
        }
        //두 시리얼 번호의 숫자 합계가 같을 경우
        else {
            //사전 순으로 정렬
            return serial1 < serial2;
        }

    }
}

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(NULL);

    int case_count;
    vector <string> vec;
    string test_case;


    cin >> case_count;

    for (int i = 0; i < case_count; i++) {
        cin >> test_case;
        vec.push_back(test_case);
    }
    //사용자 정의 함수를 이용하여 직접 지정한 방식대로 정렬
    sort(vec.begin(), vec.end(), compare);

    for (int i = 0; i < case_count; i++) {
        cout << vec[i] << endl;
    }


    return 0;
}

 

🙏 문제 풀이

sort(vec.begin(), vec.end(), compare);

• algorithm 라이브러리의 sort() 함수의 3번째 인자 값으로 사용자가 정의한 함수를 넣으면, 해당 함수의 반환 값에 따라 정렬됨.
• sort함수로 vector를 정렬하는 방법 : sort(vec.begin(), vec.end(), (사용자 정의 함수)) / 3번째 인자를 넣지 않으면 default는 오름차순