[Algorithm] 백준 1431번 : 시리얼 번호 (C++)

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

1431번: 시리얼 번호

---

[문제]

다솜이는 기타를 많이 가지고 있다. 그리고 각각의 기타는 모두 다른 시리얼 번호를 가지고 있다. 다솜이는 기타를 빨리 찾아서 빨리 사람들에게 연주해주기 위해서 기타를 시리얼 번호 순서대로 정렬하고자 한다.

모든 시리얼 번호는 알파벳 대문자 (A-Z)와 숫자 (0-9)로 이루어져 있다.

시리얼번호 A가 시리얼번호 B의 앞에 오는 경우는 다음과 같다.

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

시리얼이 주어졌을 때, 정렬해서 출력하는 프로그램을 작성하시오.

[입력]

첫째 줄에 기타의 개수 N이 주어진다. N은 50보다 작거나 같다. 둘째 줄부터 N개의 줄에 시리얼 번호가 하나씩 주어진다. 시리얼 번호의 길이는 최대 50이고, 알파벳 대문자 또는 숫자로만 이루어져 있다. 시리얼 번호는 중복되지 않는다.

[알고리즘 분류]

• 정렬

---

[해설]

이번 문제는 algorithm 헤더에서 제공되는 sort 메서드를 이용하면 쉽게 해결할 수 있는 문제였다.

문제에서 요구하는 3가지 정렬 기준을 작성하고 이를 sort 메서드에 적용하면 된다.

아래의 코드에서 작성한 정렬 기준식 c는 처음에 문자열의 길이를 비교하고 다르다면 길이 순서로, 길이가 같다면 문자열에서 숫자만을 골라 이를 a1, b1에 더해 더한 값을 기준으로 정렬을 수행한다.

(문자가 숫자인지 확인하고 더해야하기에 문자'0'과 '9'의 아스키코드 값 사이인지 확인한다. 그리고 48을 빼는 이유는 48을 빼야 해당 숫자의 값을 구할 수 있기 때문이다.)

그리고 마지막 조건문으로는 사전순으로 정렬을 수행한다.

[정답 코드]

#include <bits/stdc++.h>
using namespace std;

int c(string a,string b) {
	int aS=a.size(),bS=b.size(),a1=0,b1=0;
	if(aS!=bS) return aS < bS;
	for(int i=0;i<a.size();i++) {
	  if(a[i]>='0'&&a[i]<='9') a1+=int(a[i])-48;
	  if(b[i]>='0'&&b[i]<='9') b1+=int(b[i])-48;
	}
	if(a1!=b1) return a1 < b1;
	return a < b;
}

int main() {
	int N;
	string s[50];
	cin >> N;
	for(int i=0;i<N;i++) cin >> s[i];
	sort(s,s+N,c);
	for(int i=0;i<N;i++) cout << s[i] << "\n";
}

[결과]