7.4：移动距离不超过K的排序

最新推荐文章于 2024-03-06 14:03:13 发布

HackerTerry

最新推荐文章于 2024-03-06 14:03:13 发布

阅读量95

点赞数

分类专栏：数据结构与算法文章标签：算法数据结构 java

本文链接：https://blog.csdn.net/SCR1209/article/details/130905475

版权

数据结构与算法专栏收录该内容

145 篇文章 0 订阅

订阅专栏

该代码实现了一个使用小根堆对几乎有序数组进行排序的方法，保证每个元素移动距离不超过k。算法首先将k个最小元素放入堆中，然后依次替换堆顶元素并添加新元素，直到所有元素排序完毕，总时间复杂度为O(NlogK)。

摘要由CSDN通过智能技术生成

// 题目描述
// 已知一个几乎有序的数组。几乎有序是指，如果把数组排好顺序的话，每个元素移动的距离一定不超过k
// k相对于数组长度来说是比较小的。请选择一个合适的排序策略，对这个数组进行排序。

public class SortArrayDistanceLessK {
    public static void sortedArrDistanceLessK(int[] arr, int k) {
        if (arr == null || arr.length < 2 || k == 0) {
            return;
        }
        int index = 0;
        int i = 0;
        // 默认是小根堆
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        while (i <= Math.min(arr.length - 1, k)) {
            heap.add(arr[i]);
            i++;
        }
        while (i <= arr.length - 1) {
            arr[index] = heap.poll();
            index++;
            heap.add(arr[i]);
            i++;
        }
        while (!heap.isEmpty()) {
            arr[index] = heap.poll();
            index++;
        }
    }
}
    // 时间复杂度使NlogK