백준 1431 (C++) 시리얼 번호

 

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

1431번: 시리얼 번호

 

 

1. A와 B의 길이가 다르면, 짧은 것이 먼저 온다.
2. 만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다.
(숫자인 것만 더한다)
3. 만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.

 

 

위의 주어진 조건에 맞게 sort에 사용될 compare함수(이하 cmp)를 작성하면 된다.

다만 수의 합을 구하는 코드를 함수로 작성하고 싶어 코드-2에서 변형하였다.

 

코드 - 1

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;
using ll = long long;
int n, k;
vector<string> v;
bool cmp(string s1, string s2) {
    if (s1.length() == s2.length()) {
        int sum_s1 = 0, sum_s2 = 0;
        for (int i = 0; i < s1.length(); i++) {
            if ('0' <= s1[i] && s1[i] <= '9') {
                sum_s1 += s1[i] - '0';
            }
            if ('0' <= s2[i] && s2[i] <= '9') {
                sum_s2 += s2[i] - '0';
            }
        }
        if (sum_s1 != sum_s2) {
            return sum_s1 < sum_s2;
        }
        else return s1 < s2;
    }
    else return s1.length() < s2.length();
}
int main(void) {
    cin >> n;
    for (int i = 0; i < n; i++) {
        string s;
        cin >> s;
        v.push_back(s);
    }
    sort(v.begin(), v.end(), cmp);
    for (int i = 0; i < n; i++) {
        cout << v[i] << '\n';
    }
    return 0;
}

 

 

 

 

위와 같은 코드이나, s1과 s2의 sum이 같은지 비교하는 코드를 함수로 작성하였다.

코드 - 2 

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;
using ll = long long;
int n, k;
vector<string> v;
bool is_sum_diff(string s1, string s2, int *sum_s1, int *sum_s2) {
    for (int i = 0; i < s1.length(); i++) {
        if ('0' <= s1[i] && s1[i] <= '9') {
            *sum_s1 += s1[i] - '0';
        }
        if ('0' <= s2[i] && s2[i] <= '9') {
            *sum_s2 += s2[i] - '0';
        }
    }
    return *sum_s1 != *sum_s2;
}
bool cmp(string s1, string s2) {
    if (s1.length() == s2.length()) {
        int sum_s1 = 0, sum_s2 = 0;
        if (is_sum_diff(s1, s2, &sum_s1, &sum_s2)) {
            return sum_s1 < sum_s2;
        }
        else return s1 < s2;
    }
    else return s1.length() < s2.length();
}
int main(void) {
    cin >> n;
    for (int i = 0; i < n; i++) {
        string s;
        cin >> s;
        v.push_back(s);
    }
    sort(v.begin(), v.end(), cmp);
    for (int i = 0; i < n; i++) {
        cout << v[i] << '\n';
    }
    return 0;
}