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

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

1431번: 시리얼 번호

 

MyStr 구조체에 비교연산자 재정의

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

struct MyStr {
	string str;
	bool operator<(const MyStr& other) const {
		if (str.length() == other.str.length()) {
			int mySum = 0;
			int otherSum = 0;
			for (int i = 0; i < str.length(); i++) {
				if (str[i] >= '0' && str[i] <= '9') {
					mySum += str[i] - '0';
				}
				if (other.str[i] >= '0' && other.str[i] <= '9') {
					otherSum += other.str[i] - '0';
				}
			}
			if (mySum == otherSum) {
				return str < other.str;
			}
			return mySum < otherSum;
		}
		return str.length() < other.str.length();
	}
};

vector<MyStr> v;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int n;
	cin >> n;
	for (int i = 0; i < n; i++) {
		string str;
		cin >> str;
		v.push_back({str});
	}
	sort(v.begin(), v.end());
	for (auto a : v) {
		cout << a.str << "\n";
	}
};