Java二维数组排序

时间:2025-04-04 08:25:42

今天在刷 力扣的时候,有道题卡住了,没办法去看题解,发现题解是将其二维数组排序了,我才猛然发现我还不了解二位数组排序的知识。所以来记录一下。


答案中的代码是这样的,假设需要排序的数组intervals:

int[][] intervals = {{2,3},{2,9},{4,5},{3,7},{6,7},{8,9},{1,10}};
(intervals, new Comparator<int[]>() {
    @Override
    public int compare(int[] o1, int[] o2) {
        if(o1[0]==o2[0]){
            return o1[1] - o2[1];
        }
        return o1[0] - o2[0];
    }
});

此代码中,对于每个o1和o2数组,若各自第一个元素(也就是o1[0]和o2[0])相等,则按照各自第二个元素进行升序比较,否则就按照第一个元素进行升序比较。

在此,o1[0] - o2[0] 表示升序,o2[0] - o1[0] 表示降序。

比如这个代码,结果如下:

1 10 
2 3 
2 9 
3 7 
4 5 
6 7 
8 9 

o1[0]表示第一个元素,以此类推,所以我们想要根据第几个元素排序,就写入就好:

// 按照第三个元素排序
int[][] intervals = {{2,3,4,5},{2,9,7,5},{4,5,1,5},{3,7,1,2},{6,7,3,4}};
(intervals, new Comparator<int[]>() {
    @Override
    public int compare(int[] o1, int[] o2) {
        return o1[2] - o2[2];
    }
});

得到的结果如下:

4 5 1 5 
3 7 1 2 
6 7 3 4 
2 3 4 5 
2 9 7 5 

另外,他还有其他的写法:

1、使用Lambda表达式的方式对Comparator比较器进行简写(JDK1.8+)

int[][] intervals = {{2,3,4,5},{2,9,7,3},{4,5,1,5},{3,7,1,2},{6,7,3,4}};
(intervals, (o1, o2) -> {
    return o1[2] - o2[2];
});

2、使用()方法

static <T> Comparator<T> comparingInt(ToIntFunction<? super T> keyExtractor)
int[][] intervals = {{2,3,4,5},{2,9,7,3},{4,5,1,5},{3,7,1,2},{6,7,3,4}};
(intervals, (o -> o[2]));

下面的其他方法引用:() 实现二维数组排序