题目
一个无序数组,如果从小到大排好序,任何一个元素任何一个元素移动索引长度不超过k,实现从小到大排序
例:
k=2
排序后,[4]移动2个索引,[6]移动2个索引,[1]移动2个索引,[3]移动2个索引,[7]移动1个索引,[6]移动1个索引
解析
给定一个满足条件的数组,k=2;创建一个 大小为k+1的小根堆,先填满,出一个元素,在入一个元素即可实现排序
代码
public static void sortedArrDistanceLessK(int[] arr, int k) {
if (k == 0) {
return;
}
// 默认小根堆
PriorityQueue<Integer> heap = new PriorityQueue<>();
int index = 0;
for (; index <= Math.min(arr.length - 1, k - 1); 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();
}
}