백준 1431번 : 시리얼 번호

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

1431번: 시리얼 번호

 문제설명 

 

문자열이 주어졌을경우 조건에 맞춰서 정렬 하는문제.

 

 알고리즘 

 

1. sort 함수에서 비교 함수를 정의하여 정렬한다 

   - 길이가 짧은 것이 앞으로

   - 길이가 같다면 문자열에 있는 숫자를 더해서 작은게 앞으로 

   - 더한 값도 같다면 사전순으로 정렬

 

 코드 

#include<iostream>
#include<vector>
#include<algorithm>
#include<string>
 
using namespace std;
 
int getSum(string a) {
    int n = a.length(), sum = 0;
    for (int i = 0; i < n; i++) {
        if (a[i] - '0' <= 9 && a[i] - '0' >= 0) {
            sum += a[i] - '0';
        }
    }
    return sum;
}
 
bool compare(string str1, string str2)
{
    //길이가 짧은것이 우선
    if (str1.length() != str2.length()) {
        return str1.length() < str2.length();
    }    
    //길이가 같을때
    else {
        if (getSum(str1) != getSum(str2)) {
            return getSum(str1) < getSum(str2);
        }
        else return str1 < str2;
    }
 
}
vector<string> v;
int main() {
    int n;
    cin >> n;
    string str;
    for (int i = 0; i < n; i++) {
 
        cin >> str;
        v.push_back(str);
    }
 
    sort(v.begin(), v.end(),compare);
 
    for (int i = 0; i < n; i++)cout << v[i]<<"\n";
}