BOJ_1431 : 시리얼 번호

👉 문제링크

1431번: 시리얼 번호

---

---

🔸 문제 분석 🔸

• 문자열 배열을 다음과 같은 세 조건으로 정렬해서 출력한다.

1. 길이가 다르다면 짧은 것이 앞에 온다.
2. 길이가 같다면, 한 문자열 안에 들어있는 숫자의 합이 작은 것이 먼저 온다.
3. 숫자의 합도 같다면, 문자열을 사전순 비교한다.

🔸 코드 🔸

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        String[] arr = new String[n];
        for (int i = 0; i < n; i++) arr[i] = br.readLine();

        Arrays.sort(arr, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                int len1 = o1.length();
                int len2 = o2.length();
                if (len1 != len2)
                    return len1 - len2;
                else {
                    int sum1 = 0, sum2 = 0;
                    for (int i = 0; i < len1; i++){
                        if (o1.charAt(i) >= '0' && o1.charAt(i) <= '9')
                            sum1 += o1.charAt(i) - '0';
                        if (o2.charAt(i) >= '0' && o2.charAt(i) <= '9')
                            sum2 += o2.charAt(i) - '0';
                    }

                    if (sum1 != sum2)
                        return sum1 - sum2;
                    else
                        return o1.compareTo(o2);
                }
            }
        });

        for (String s : arr) {
            System.out.println(s);
        }
    }
}

🔸 코드 해석 🔸

• String 배열 arr에 문자열들을 입력받고 Arrays.sort()를 사용해 정렬한다.
  • 정렬 조건을 orverride로 재정의했다. (음수 반환 == o1이 빠르다)
    • 두 문자열의 길이가 다르면  짧은게 더 빠르다
    • 두 문자열에 포함되어 있는 숫자들의 총 합이 작은쪽이 더 빠르다.
    • 위 두 조건으로도 판별이 안나면 compareTo()를 사용해 사전식 정렬한다.

---

🔸 end 🔸

• 이번 정렬 3문제 풀기는 Arrays.sort()의 정렬 조건을 다양하게 바꿔보는 좋은 연습이 된 것 같다.
• 클래스를 따로 만들어 다양한 인자로 정렬하는 방법과 이 문제처럼 내장형으로 메소드 오버라이드를 이용해 정렬 조건을 바꾸는 방법을 확실히 익혀가는 것 같다.
  • 내장형으로 정렬조건을 바꿀때는 정렬 인자가 1개로 똑같을 때만 가능한 것 같다.