堆排序拓展应用

题目：

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

思路：

        由于数组是几乎有序的，假如在第k+1个数之后存在最小值，那么元素移动到第一个位置需要移动超过k次，不符合，所以最小值必在前k+1个数里，将前k个值先放入小根堆中，将第k+1个数存入小根堆中，此时的堆顶的元素就是前K+1个数中的最小值，将堆顶元素取出，此时堆中只有k个值，继续循环，直到将所有数都存入数组。最后依次将最小堆中堆顶数值弹出

public static void sortedArrDistanceLessK(int[] arr, int k) {
    if (k == 0) {
        return;
    }
    //默认小根堆
    PriorityQueue<Integer> heap = new PriorityQueue<>();
    int index = 0;
    //将前k个值放入小根堆中
    for (; index <= Math.min(arr.length - 1, k - 1); index++) {
        heap.add(arr[index]);
    }
    int i = 0;
    //由于数组是几乎有序的，假如在第k+1个数之后存在最小值
    //那么元素移动到第一个位置需要移动超过k次，不符合，所以最小值必在前k+1个数里
    for (; index < arr.length; i++, index++) {
        //将第k+1个数存入小根堆中，此时的堆顶的元素就是前K+1个数中的最小值
        heap.add(arr[index]);
        //将堆顶元素取出，此时堆中只有k个值，继续循环，直到将所有数都存入数组
        arr[i] = heap.poll();
    }
    //依次将最小堆中堆顶数值弹出
    while (!heap.isEmpty()){
        arr[i++]=heap.poll();
    }
}