有苦有乐的算法 --- 一个无序数组,如果从小到大排好序,任何一个元素任何一个元素移动索引长度不超过k,实现排序

题目

一个无序数组,如果从小到大排好序,任何一个元素任何一个元素移动索引长度不超过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();
	}
}
posted @ 2022-03-07 18:03  叕叕666  阅读(46)  评论(0)    收藏  举报