[백준] S3 1431번 시리얼 번호 (JAVA)

문제 출처 - Baekjoon Online Judge

문제는 여기

1431번: 시리얼 번호

[문제] 

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

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

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

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

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

[입력]

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

[출력]

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

 

 

---

[풀이]

1. Comparator 혹은 Comparable을 사용해 compare를 재정의 해서 사용해야 하는 문제였다.

2. 문자열로 입력을 받아서 처리할 수 있기에 Comparable을 사용해 문제를 풀었다.

3. 길이를 먼저 비교해 짧은 것을 우선해주었다.

4. 길이가 같은 경우 문자열에 있는 숫자들의 합을 비교해서 정렬해주었다.

5. 숫자의 합도 같다면 사전순으로 정렬해주었다.

6. 정렬이 된 결과를 출력해주었다.

[접근]

1. 객체로 받지 않고 문자열만으로 비교를 할 수 있기 때문에 Comparator를 사용해서 처리했다.

2. compare를 조건에 맞게 재정의해주었다

[코드]

package BOJ_silver;

import java.util.*;
import java.io.*;

public class Main_S3_1431 {
	static int n;
	static String[] serial;

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		n = Integer.parseInt(br.readLine());
		
		serial = new String[n];
		
		// 시리얼 번호 입력받기
		for (int i = 0; i < n; i++) {
			serial[i] = br.readLine();
		}
		
		Arrays.sort(serial, new Comparator<String>(){
			@Override
			public int compare(String o1, String o2) {
				if (o1.length() < o2.length()) // 길이가 짧은 것을 오름차순 정렬
					return -1;
				else if (o1.length() == o2.length()) {
					if (sum(o1) == sum(o2)) { // 합이 같은 경우
						return o1.compareTo(o2); // 둘을 비교해서 사전순으로 정렬
					}
					else { // 합이 다른 경우
						return Integer.compare(sum(o1), sum(o2)); // 오름차순 정렬
					}
				}
				else
					return 1;
			}
		});
		
		for (int i = 0; i < n; i++) {
			System.out.println(serial[i]);
		}
	}
	
	// 시리얼 번호에서 숫자로 이루어진 경우 해당 값들의 합을 구하는 방식
	public static int sum(String str) {
		int sum = 0;
		
		for (int i = 0; i < str.length(); i++) {
			if (str.charAt(i) >= '0' && str.charAt(i) <= '9')
				sum += str.charAt(i) - '0';
		}
		
		return sum;
	}
}