Java利用小顶堆解决TopK问题

最新推荐文章于 2024-05-31 21:51:47 发布

比尔咖啡

最新推荐文章于 2024-05-31 21:51:47 发布

阅读量314

点赞数 1

文章标签： java 数据结构

本文链接：https://blog.csdn.net/qq_38453831/article/details/123334223

版权

本文通过自己动手，使用Java实现了小顶堆解决TopK问题，以清晰的思路整理了整个过程，适合学习巩固数据结构知识。

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

网上教程看着很乱，自行利用小顶堆写了一个TopK问题来练练手，仅供参考。

/**
 * TopK问题，利用小顶堆解决
 */
public class Test14 {
    public static void main(String[] args) {
        int[] arr = {9,6,8,7,0,1,10,4,2};
        int k = 5;
        int[] topK = new int[k];
        // 把数组的前k个数存到一个新数组中
        for (int i = 0; i < topK.length; i++) {
            topK[i] = arr[i];
        }
        // 利用k长度的新数组建立一个MinHeap实例
        MinHeap heapK = new MinHeap(topK);
        // 调用小顶堆方法
        heapK.minHeap();
        // 如果N-K开始到N的数中有大于堆顶元素的就替换，没有就忽略
        for(int i = k; i < arr.length; i++){
            if(arr[i] > heapK.getRoot()){
                heapK.setRoot(arr[i]);
            }
        }
        System.out.println(Arrays.toString(arr));
        System.out.println(Arrays.toString(topK));
    }
}
class MinHeap{
    private int[] data;

    public MinHeap(){

    }
    public MinHeap(int[] data){
        this.dat