[백준/C++] 1431번 시리얼 번호

---

#include <iostream>
#include <iomanip>
#include <string>
#include <vector>
#include <numeric>
#include <math.h>
#include <algorithm>
#include <functional>
using namespace std;

bool compare(const string &a, const string &b) {
    if(a.length() != b.length()) return a.length() < b.length();
    else {
        int sum_A = 0;
        int sum_B = 0;
        for(int i = 0; i < a.length(); i++) {
            if(isdigit(a[i])) sum_A += (a[i] - '0');
            if(isdigit(b[i])) sum_B += (b[i] - '0');
        }
        return sum_A == sum_B ? a < b : sum_A < sum_B;
    }
}

int main() {
    int N;
    cin >> N;
    vector<string> arr(N);
    string temp = "";

    for(int i = 0; i < N; i++) {
        cin >> temp;
        arr[i] = temp;    
    }
    sort(arr.begin(), arr.end(), compare);
    
    for(string it : arr)
        cout << it << "\n";
}

---

솔팅 할 때 compare 함수를 살짝 손봤습니다.

길이가 같지 않다면 길이비교를, 아니면 숫자만 골라서 숫자 합의 결과를 그것도 같다면 그냥 사전 순으로 비교하였습니다.

---

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

1431번: 시리얼 번호