[백준] 1431번 시리얼 번호

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

1431번: 시리얼 번호

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

bool cmp(string a, string b) {
	if (a.size() != b.size()) {
		return a.size() < b.size();
	}
	else {
		int sumOfA = 0;
		int sumOfB = 0;
		for (int i = 0; i < a.size(); i++) {
			if (a[i] < 58) {
				sumOfA += (a[i] - '0');
			}
		}
		for (int i = 0; i < b.size(); i++) {
			if (b[i] < 58) {
				sumOfB += (b[i] - '0');
			}
		}
		if (sumOfA != sumOfB) {
			return sumOfA < sumOfB; //sumOfB
		}
		else {
			return a < b;
		}
	}
}

int main() {
	cin.tie(NULL);
	ios_base::sync_with_stdio(false);
	int n;
	cin >> n;
	vector<string> v(n);
	for (int i = 0; i < n; i++) {
		cin >> v[i];
	}
	sort(v.begin(), v.end(), cmp);
	for (int i = 0; i < n; i++)
		cout << v[i] << '\n';
	return 0;
}