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

www.acmicpc.net/problem/1431

1431번: 시리얼 번호

정렬 유형의 문제입니다.

길이, 숫자 자리수의 합, 사전순대로 비교를 하기 위해서 cmp 함수를 따로 만들어줬습니다. 그리고 숫자 자리수의 합을 기준으로 정렬하기 위해서 자리수의 합을 구하기 위한 함수로 getSum 함수를 만들었습니다.

 

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

int getSum(string str) {
	int sum = 0;
	for(int i = 0; i < str.size(); i++) {
		if(str[i] >= '0' && str[i] <= '9') sum += (str[i] - '0');
	}
	
	return sum;
}

int cmp(string a, string b) {
	if(a.size() == b.size()) { 
		if(getSum(a) == getSum(b)) return a < b; // 사전순 비교
		 
		return getSum(a) < getSum(b); // 숫자 자리수의 합 비교 
	}
	
	return a.size() < b.size(); // 길이 비교  
}

int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	
    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";	
	}
}