17370845950

如何在 Java 中对二维数组按多列(主次优先级)排序

本文介绍使用 `arrays.sort()` 配合 `comparator` 链式调用,对 int 类型二维数

组按第一列升序为主、第二列升序为辅进行稳定排序的完整方法,并附可运行示例与关键注意事项。

在 Java 中,对二维数组(如 int[][])按多个维度排序,核心思路是将每行视为一个整体元素,并通过自定义 Comparator 定义复合排序规则。推荐方式是利用 Comparator.comparingInt() 进行主键比较,再通过 thenComparing() 添加次级比较逻辑——这不仅代码简洁,而且语义清晰、性能高效。

以下是一个完整可运行示例:

import java.util.*;

public class MultiColumnSort {
    public static void main(String[] args) {
        int[][] arr = {
            {1, 7},
            {2, 6},
            {3, 9},
            {4, 1},
            {5, 3},
            {6, 7}
        };

        // 按第0列(col1)升序为主,第1列(col2)升序为辅
        Arrays.sort(arr, Comparator.comparingInt(row -> row[0])
                                   .thenComparing(row -> row[1]));

        System.out.println(Arrays.deepToString(arr));
        // 输出: [[1, 7], [2, 6], [3, 9], [4, 1], [5, 3], [6, 7]]
    }
}

关键说明:

  • Comparator.comparingInt(...) 显式指定泛型类型,避免类型推断失败(尤其在旧版 JDK 中);
  • row[0] 和 row[1] 分别代表每行的首列与次列值,适用于所有非空且列数 ≥2 的 int[][];
  • 若需降序,可替换为 comparingInt(...).reversed() 或 thenComparingInt(...).reversed(),例如:
    comparingInt(row -> row[0]).thenComparing(row -> row[1]).reversed() // 全局逆序
    comparingInt(row -> row[0]).thenComparingInt(row -> -row[1])       // 仅 col2 降序

⚠️ 注意事项:

  • 确保数组中每行长度 ≥2,否则运行时抛出 ArrayIndexOutOfBoundsException;
  • Arrays.sort() 是原地排序(in-place),直接修改原数组;如需保留原始顺序,请先 clone();
  • 该方法仅适用于引用类型数组(int[][] 是 int[] 对象的数组),不适用于基本类型一维数组(如 int[])。

掌握这种链式 Comparator 写法,不仅能处理二维数组多级排序,还可轻松扩展至三列及以上(连续调用 thenComparing),是 Java 函数式排序的典型实践。