堆排序扩展题目（N*logK）

---

堆排序扩展题目

已知一个几乎有序的数组，几乎有序是指，如果把数组排好顺序的话，每个元 素移动的距离可以不超过k，并且k相对于数组来说比较小。请选择一个合适的 排序算法针对这个数据进行排序

题意的理解

显然是要我们找一个高效的算法，使数组排上序。我们选择了堆排序 开一个k大小的堆，然后不断的加入原数组的值，并且每加入一个值就弹出一个堆顶（最小值）复制到原数组。因为只涉及弹出弹入。我们选择优先队列

---

思路

1.复制前k项

int index = 0;
		for (; index < Math.min(arr.length, k); index++) {
			heap.add(arr[index]);
		}//用Math。min是为了防止k大于数组长度

2.不断输入与弹出

		int i = 0;
		for (; index < arr.length; i++, index++) {
			heap.add(arr[index]);  //从恰好堆长度在数组中的后面输入
			arr[i] = heap.poll();  //将堆顶复制到数组前端没排好的部分

3.当已经达到数组边缘时，不在输入，只弹出

		while (!heap.isEmpty()) {
			arr[i++] = heap.poll();
		}

注意i是要做外部变量，时刻指向已经排好队部分的前面一个

---

代码

import java.util.PriorityQueue;

public class Code04_SortArrayDistanceLessK {

	public void sortedArrDistanceLessK(int[] arr, int k) {
		PriorityQueue<Integer> heap = new PriorityQueue<>();
		int index = 0;
		for (; index < Math.min(arr.length, k); index++) {
			heap.add(arr[index]);
		}
		int i = 0;
		for (; index < arr.length; i++, index++) {
			heap.add(arr[index]);
			arr[i] = heap.poll();
		}
		while (!heap.isEmpty()) {
			arr[i++] = heap.poll();
		}
	}
}