10814번 : 나이순 정렬

 

https://www.acmicpc.net/problem/10814

10814번: 나이순 정렬

 

 

문제

 

 

코드

import java.io.*;
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));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int n = Integer.parseInt(br.readLine());
        
        String srr[][] = new String[n][2];
        for(int i=0;i<n;i++) {
            String str[] = br.readLine().split(" ");
            srr[i][0] = str[0];
            srr[i][1] = str[1];
        }
        
        Arrays.sort(srr, new Comparator<String[]>() {
            @Override
            public int compare(String[] o1, String[] o2) {
                int a = Integer.parseInt(o1[0]);
                int b = Integer.parseInt(o2[0]);
                return a-b;
            }
        });
        
        for(int i=0;i<n;i++) {
            System.out.println(srr[i][0] + " " + srr[i][1]);
        }
    }
}

 

 

해결방법

나이를 기준으로 정렬하면 되니 Arrays.sort()를 이용하였다. Comparator를 재정의하여 풀었는데 테스트 값은 잘 돌아갔지만 런타임 에러 (IllegalArgument)가 발생했다. 뭐가 잘못된 것인지 알아봤는데 compare(a, a)처럼 같은 값을 비교한다면 항상 0을 반환해주어야 IllegalArgument가 발생하지 않는다고 나온다. 처음 코드를 짤 땐 -1 또는 1이 반환되게 하여서 컴파일 오류가 난 것이다. 잘 정리를 해둬야겠다.