자바 - 2차원 배열 정렬 (오름차순, 내림차순, 다중 조건)

자바의 2차원 배열을 Arrays.sort() 를 통해 곧바로 정렬하려고 하면 java.lang.ClassCastException: I cannot be cast to java.lang.Comparable 오류가 발생한다. 오류 문구와 같이 비교 기준이 없어 캐스팅에 실패하기 때문이다. 

 

예시와 같이 Comparable, Comparator 인터페이스를 구현하여 정렬 기준을 추가해줘야 한다.

 

1. 오름차순, 내림차순 정렬

int[][] arr = new int[][]{{1, -1}, {2, 2}, {3, 3}, {1, 2}, {0, 4}}

Arrays.sort(arr, new Comparator<int[]>() {
    @Override
    public int compare(int[] o1, int[] o2) {
        return o1[0] - o2[0]; // 첫번째 숫자 기준 오름차순 {0, 4}, {1, 2}, {1, -1}, {2, 2}, {3, 3}
        return o2[0] - o1[0]; // 첫번째 숫자 기준 내림차순 {3, 3}, {2, 2}, {1, 2}, {1, -1}, {0, 4}
        return o1[1] - o2[1]; // 두번째 숫자 기준 오름차순 {1, -1}, {2, 2}, {1, 2}, {3, 3}, {0, 4} 
        return o2[1] - o1[1]; // 두번째 숫자 기준 내림차순 {0, 4}, {3, 3}, {1, 2}, {2, 2}, {1, -1} 
    }
});

 

2. 다중 조건 정렬

Arrays.sort(arr, new Comparator<int[]>() {
    @Override
    public int compare(int[] o1, int[] o2) {
        if (o1[1] == o2[1]) {      // 두번째 숫자가 같으면
            return o1[0] - o2[0];  // 첫번째 숫자가 증가하는 순서로,
        } else {                   // 그렇지 않으면
            return o1[1] - o2[1];  // 두번째 숫자가 증가하는 순서로 정렬한다.
        }                          // {1, -1}, {1, 2}, {2, 2}, {3, 3}, {0, 4}
    }
});