본문 바로가기

백준/C++

[Baekjoon/C++] 1431번 - 시리얼 번호

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

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

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

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

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

 

입력

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

 

출력

첫째 줄부터 차례대로 N개의 줄에 한줄에 하나씩 시리얼 번호를 정렬한 결과를 출력한다.

 


예제 입력 예제 출력
5
ABCD
145C
A
A910
Z321
A
ABCD
Z321
145C
A910
2
Z19
Z20
Z20
Z19
4
34H2BJS6N
PIM12MD7RCOLWW09
PYF1J14TF
FIPJOTEA5
FIPJOTEA5
PYF1J14TF
34H2BJS6N
PIM12MD7RCOLWW09
5
ABCDE
BCDEF
ABCDA
BAAAA
ACAAA
ABCDA
ABCDE
ACAAA
BAAAA
BCDEF

풀이

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

int sum(string s);
bool compare(string a, string b);

// 모든 자리수의 합 구하기
int sum(string s) {
	int n = 0;

	for (int i = 0; i < s.length(); i++)
		if (!(s[i] >= 'A' && s[i] <= 'Z')) n += s[i] - '0';

	return n;
}

bool compare(string a, string b) {
	if (a.length() == b.length()) {
		int n1 = sum(a);
		int n2 = sum(b);

		// 사전순으로 앞서는 것
		if (n1 == n2) return a < b;

		// 작은 합을 가지는 것
		return n1 < n2;
	}

	// 길이가 짧은 것
	return a.length() < b.length();
}

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

	int N;
	cin >> N;

	string arr[51];

	for (int i = 0; i < N; i++)
		cin >> arr[i];

	sort(arr, arr + N, compare);

	for (int i = 0; i < N; i++)
		cout << arr[i] << "\n";

	return 0;
}

원하는 방식으로 시리얼번호를 정렬하기 위해서 compare 함수를 정의해서 문제를 풀었다.

 

// 모든 자리수의 합 구하기
int sum(string s) {
	int n = 0;

	for (int i = 0; i < s.length(); i++)
		if (!(s[i] >= 'A' && s[i] <= 'Z')) n += s[i] - '0';

	return n;
}

bool compare(string a, string b) {
	if (a.length() == b.length()) {
		int n1 = sum(a);
		int n2 = sum(b);

		// 사전순으로 앞서는 것
		if (n1 == n2) return a < b;

		// 작은 합을 가지는 것
		return n1 < n2;
	}

	// 길이가 짧은 것
	return a.length() < b.length();
}

2번 조건을 위해 시리얼번호의 숫자만을 더해서 return하는 sum()을 구현했다. 각 자리의 문자가 A와 Z 사이가 아니라면 '0'을 뺀 값을 n에 더하도록 하였다.

 

compare 함수 내부에서는 가장 까다로운 3번 조건이 가장 먼저 확인될 수 있도록 if문을 배치했다.