백준 1431번 시리얼 번호 (자바) 풀이

시간 제한메모리 제한제출정답맞힌 사람정답 비율

2 초

128 MB

11139

6005

4941

55.133%

문제

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

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

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

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

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

입력

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

출력

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

예제 입력 1 복사

5
ABCD
145C
A
A910
Z321

예제 출력 1 복사

A
ABCD
Z321
145C
A910

예제 입력 2 복사

2
Z19
Z20

예제 출력 2 복사

Z20
Z19

예제 입력 3 복사

4
34H2BJS6N
PIM12MD7RCOLWW09
PYF1J14TF
FIPJOTEA5

예제 출력 3 복사

FIPJOTEA5
PYF1J14TF
34H2BJS6N
PIM12MD7RCOLWW09

예제 입력 4 복사

5
ABCDE
BCDEF
ABCDA
BAAAA
ACAAA

예제 출력 4 복사

ABCDA
ABCDE
ACAAA
BAAAA
BCDEF

 

 

풀이

 

굉장히 친절한 문제이다

 

문제에 어떻게 정렬을 하라고 다 알려주고 있다.

 

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

단순히 조건문으로 A와 B의 길이를 비교해 정렬하면 된다.

2 -> A와 B의 길이가 같다면 문자열 내 모든 숫자의 합 순으로 정렬한다.

마찬가지로 조건문안에 반복문을 삽입하면 된다.

3 -> 두 방법으로도 정렬이 안된다면 단순 사전순으로 정렬한다.

 

3번째 방법이 문제이다.

두 방법으로도 정렬이 안된다면? 이 어떤 상황일까?

단순하다. A와 B의 길이도 같은데 문자열 내 모든 숫자의 합마저 같다면 이라는 조건이 되어버린다.

 

단순히 2번의 조건문 안에 숫자의 합을 비교하는 조건문을 추가해주면 된다.

 

package baekjoon.a1431;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

public class a1431 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        List<String> list = new ArrayList<>();
        for (int i = 0; i < N; i++) {
            list.add(br.readLine());
        }
        list.sort((o1, o2) -> {
            if (o1.length() != o2.length()) {
                return o1.length() - o2.length();
            } else {
                int a = 0;
                int b = 0;
                for (int i = 0; i < o1.length(); i++) {
                    int num1 = o1.charAt(i) - '0';
                    int num2 = o2.charAt(i) - '0';
                    if (num1 > 0 && num1 < 10) {
                        a += num1;
                    }
                    if (num2 > 0 && num2 < 10) {
                        b += num2;
                    }
                }
                if (a == b) {
                    return o1.compareToIgnoreCase(o2);
                }
                return a - b;
            }
        });
        list.forEach(System.out::println);
    }
}