SortArrayDistanceLessK

最新推荐文章于 2023-09-10 11:03:31 发布

原创最新推荐文章于 2023-09-10 11:03:31 发布 · 334 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#java #数据结构 #算法

收录于

Algorithmic problem 专栏收录该内容

3 篇文章

订阅专栏

这篇博客介绍了一种针对几乎有序数组的排序算法，利用优先级队列(PriorityQueue)来实现。在数组中，元素的移动距离不超过K，K相对较小。算法思路是初始化一个大小为K+1的优先级队列，然后逐步添加数组元素并移除最小值，确保排序过程中的元素移动不超过K次。源代码展示了具体的实现细节。

[Question]

已知一个几乎有序的数组，几乎有序是指如果把数组排好序的话，每个元素移动的距离可以不超过K，并且K相对于数组来说比较小。

[Idea]

使用优先级队列结构 PriorityQueue length=K+1
首先将数组前"K+1"个数add进 PriorityQueue 
其次循环 add 数组下一个数 并 poll up 数组第一个数

[Source code]

import java.util.PriorityQueue;

public class code04_SortArrayDistanceLessK {

    public static void sortArrayDistanceLessK(int[] array, int k) {
        if (array == null || array.length < 2) {
            return;
        }
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        int index = 0;
        for (; index < Math.min(array.length - 1, k); index++) {
            heap.add(array[index]);
        }
        int i = 0;
        for (; index < array.length; i++, index++) {
            heap.add(array[i]);
            array[i] = heap.poll();
        }
        while (!heap.isEmpty()) {
            array[i++] = heap.poll();
        }

    }
}