백준 1431 시리얼 번호

문제

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

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

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

1. A와 B의 길이가 다르면, 짧은 것이 먼저 온다.

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

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

---

코드

import java.io.*;
import java.util.Comparator;
import java.util.PriorityQueue;

class Main{
    public static class Serial{
        public int size;
        public int sum;
        public String serial;

        public Serial(String serial){
            this.size=serial.length();
            this.serial=serial;

            int temp=0;
            for(int i=0; i<serial.length(); i++){
                int n=Character.getNumericValue(serial.charAt(i));
                if(n>=0&&n<10)
                    temp+=n;
            }
            this.sum=temp;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));

        int n=Integer.parseInt(br.readLine());
        PriorityQueue<Serial> pq=new PriorityQueue<>(new Comparator<Serial>() {
            @Override
            public int compare(Serial o1, Serial o2) {
                if(o1.size<o2.size)
                    return -1;
                else if(o1.size==o2.size){
                    if(o1.sum<o2.sum)
                        return -1;
                    else if(o1.sum==o2.sum){
                        return o1.serial.compareTo(o2.serial);
                    }
                    else
                        return 1;
                }
                else
                    return 1;
            }
        });
        for(int i=0; i<n; i++){
            Serial serial=new Serial(br.readLine());
            pq.add(serial);
        }
        for(int i=0; i<n; i++)
            bw.write(pq.remove().serial+"\n");

        bw.flush();
    }
}

---

해결 과정

1. 자바의 문법에 대해서 배우는 시간이였다. Serial 클래스를 만들어서 Priority Queue에 넣을려고 했는데 Comparator 인터페이스를 사용해서 비교해야 했다. 문제에서 주어진 조건대로, 문자열 길이, 숫자의 합, 사전식 순서대로 compare 메소드를 오버라이딩 해줬다.
2. 😁