用小根堆合并k个长度相等的有序数组

自定义一个HeadNode类，两个字段分别代表 在哪个数组 和 在该数组的那个位置。

public static void main(String[] args) {
    int[][] mat = new int[][]{{1,2,4,6}, {2,4,5,8}, {1,5,6,7}};
    int[] res = mergeOrderedArray(mat);
    System.out.println(Arrays.toString(res));
}
public static int[] mergeOrderedArray(int[][] mat) {
    int k = mat.length; // 数组的个数
    int len = mat[0].length; // 每个数组的长度
    if (k == 0) return new int[0];
    if (k == 1) return mat[0];

    int[] res = new int[k * len]; // 存放合并后的数据

    // 按节点所对应到的数组的值进行比较
    PriorityQueue<HeadNode> minHeap =
            new PriorityQueue<>(Comparator.comparingInt(n -> mat[n.arraysIndex][n.index]));

    // 创建所有数组第一个元素的代表节点，然后加入堆中
    for (int i = 0; i < k; i++) minHeap.add(new HeadNode(i, 0));

    int count = 0;
    while (count < k * len) {
        HeadNode node = minHeap.poll();
        res[count++] = mat[node.arraysIndex][node.index]; // 将该节点在数组中代表的值存入res中
        if (node.index < len - 1) {
            // 将之前poll节点在数组中的后一个节点存入堆中
            minHeap.add(new HeadNode(node.arraysIndex, node.index+1));
        }
    }
    return res;
}
private static class HeadNode {
    int arraysIndex; // 代表数组编号
    int index; // 当前数组的当前value的下标
    public HeadNode (int arraysIndex, int index) {
        this.arraysIndex = arraysIndex;
        this.index = index;
    }
}