백준 1431번: 시리얼 번호 with Java

시리얼 번호

 

 

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

2 초

128 MB

20385

11612

9402

57.851%

문제

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

모든 시리얼 번호는 알파벳 대문자 (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. 길이 짧으면 무조건 앞

2. 길이 같으면 숫자들만 더해서 작은 게 먼저(숫자가 없으면 0이니까 영어만 있는 게 앞)

3. 2번으로도 같으면 숫자가 알파벳보다 앞임...

 

아

정렬 문제 한 번 더 풀라고 이거 골랐는데 갑자기 문제를 바꾸고 싶고.. 마음이 힘들고 어

그래도 해야지 난... 상여자니까ㅅ

 

다솜아 언니가 기타 정리 깔쌈하게.해주꼐.

 

 

Collections.sort를 이용했다. compare 재정의함

그냥 저 문제 조건인 

1. 길이 짧으면 무조건 앞

2. 길이 같으면 숫자들만 더해서 작은 게 먼저(숫자가 없으면 0이니까 영어만 있는 게 앞)

3. 2번으로도 같으면 숫자가 알파벳보다 앞임...

 

이걸 정직하게 구현함

이쯤에서 알아보는 compare 메소드

 

compare 메소드

Comparator 인더페이스의 메소드로, 두 객체를 비교하여 정렬 순서를 정의하는 데 사용된다.

compare(T o1, T o2) 가 있을 때 Comparator가 비교해서 세 종류의 반환값을 뱉는다. 

0: 동일할 때

-1: o2가 o1보다 클 때 o1 - o2 순으로 정렬

1: o1이 o2보다 클 때 o2 - o1 순으로 정렬

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
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());
        ArrayList<String> list = new ArrayList<>();
        for (int i = 0; i < N; i++) {
            list.add(br.readLine());
        }

        Collections.sort(list, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                if (s1.length() > s2.length()) {
                    return 1;
                } else if (s1.length() < s2.length()) {
                    return -1;
                }

                int s1Sum = 0;
                int s2Sum = 0;
                for (int i = 0; i < s1.length(); i++) {
                    if ('0' <= s1.charAt(i) &&'9' >= s1.charAt(i)) {
                        s1Sum += s1.charAt(i) - '0';
                    }

                    if ('0' <= s2.charAt(i) && '9' >= s2.charAt(i)) {
                        s2Sum += s2.charAt(i) - '0';
                    }
                }

                if (s1Sum > s2Sum) {
                    return 1;
                } else if (s1Sum < s2Sum) {
                    return -1;
                }

                for (int i = 0; i < s1.length(); i++) {
                    if (s1.charAt(i) > s2.charAt(i)) {
                        return 1;
                    } else if (s1.charAt(i) < s2.charAt(i)) {
                        return -1;
                    }
                }

                return 0;
            }
        });

        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < N; i++) {
            sb.append(list.get(i)).append("\n");
        }

        System.out.println(sb);
    }
}

 

 

 

 

어라 이게 되네...?

시간 복잡도 때문에 걸리나 했는데 그렇게 많지 않아서 그런가? 통과다~

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
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());
        ArrayList<String> list = new ArrayList<>();
        for (int i = 0; i < N; i++) {
            list.add(br.readLine());
        }   // 입력 받고 list에 저장
        
        Collections.sort(list, (s1, s2) -> {
            if (s1.length() > s2.length()) {
                return 1;
            } else if (s1.length() < s2.length()) {
                return -1;
            }   // 비교해서 문자 길이가 더 길면 뒤로 보내기

            int s1Sum = 0;
            int s2Sum = 0;  // sum 변수 0으로 각각 지정
            for (int i = 0; i < s1.length(); i++) {
                if ('0' <= s1.charAt(i) &&'9' >= s1.charAt(i)) {
                    s1Sum += s1.charAt(i) - '0';
                }

                if ('0' <= s2.charAt(i) && '9' >= s2.charAt(i)) {
                    s2Sum += s2.charAt(i) - '0';
                }
            }   // 문자열에 0~9 포함돼있으면 0으로 초기화한 sum 변수에 더해주기

            if (s1Sum > s2Sum) {
                return 1;
            } else if (s1Sum < s2Sum) {
                return -1;
            }   // 숫자 더 크면 뒤로

            for (int i = 0; i < s1.length(); i++) {
                if (s1.charAt(i) > s2.charAt(i)) {
                    return 1;
                } else if (s1.charAt(i) < s2.charAt(i)) {
                    return -1;
                }
            }   // 전부 다 같을 때는 사전순 정렬

            return 0;
        });

        StringBuilder sb = new StringBuilder(); // 출력 위한 스트링빌더
        for (int i = 0; i < N; i++) {
            sb.append(list.get(i)).append("\n");
        }   // 출력 형식 맞춰 더해주기

        System.out.println(sb); // 출력
    }
}

 

주석도 써줬다! 끝