合并Java中的K个排序数组

这是一个经典的面试问题。
通过使用堆可以解决此问题。时间复杂度为O（nlog（k）），其中n是元素总数，而k是数组数。
将O（log（k））插入到堆中需要花费O（log（k））来删除最小元素。
class ArrayContainer implements Comparable {
int[] arr;
int index;

public ArrayContainer(int[] arr, int index) {
	this.arr = arr;
	this.index = index;
}

@Override
public int compareTo(ArrayContainer o) {
	return this.arr[this.index] - o.arr[o.index];
}}

public class KSortedArray {
public static int[] mergeKSortedArray(int[][] arr) {
//PriorityQueue is heap in Java
PriorityQueue queue = new PriorityQueue();
int total=0;

	//add arrays to heap
	for (int i = 0; i < arr.length; i++) {
		queue.add(new ArrayContainer(arr[i], 0));
		total = total + arr[i].length;
	}

	int m=0;
	int result[] = new int[total];

	//while heap is not empty
	while(!queue.isEmpty()){
		ArrayContainer ac = queue.poll();
		result[m++]=ac.arr[ac.index];

		if(ac.index < ac.arr.length-1){
			queue.add(new ArrayContainer(ac.arr, ac.index+1));
		}
	}

	return result;
}

public static void main(String[] args) {
	int[] arr1 = { 1, 3, 5, 7 };
	int[] arr2 = { 2, 4, 6, 8 };
	int[] arr3 = { 0, 9, 10, 11 };

	int[] result = mergeKSortedArray(new int[][] { arr1, arr2, arr3 });
	System.out.println(Arrays.toString(result));
}}

最后，开发这么多年我也总结了一套学习Java的资料与面试题，如果你在技术上面想提升自己的话，可以关注我，私信发送领取资料或者在评论区留下自己的联系方式，有时间记得帮我点下转发让跟多的人看到哦。