TopK问题的解析及代码

问题描述:

从数组arr[1…n]中,找出最大的K个数

方法一:堆排序

排序算法参考:十大排序算法总结

思路

只找到topk,不排序topk。

  1. 先用前k个元素生成一个小顶堆(即父节点比子节点小的完全二叉树),这个小顶堆用于存储当前最大的k个元素。
  2. 接着,从第k+1个元素开始扫描,和堆顶(堆中最小元素)进行比较,如果被扫描的元素大于堆顶则替换堆顶的元素,并调整堆,以保证堆内的k个元素总是当前最大的k个元素。
  3. 扫描完所有n-k个元素,最终堆中的k个元素就是topk

采用优先队列:

下面的代码的目的是利用优先队列得到第k大元素

import java.util.PriorityQueue;
public class Solution{
   
    public int findKthLargest(int[] nums, int k){
   
        int len = nums.length;
        
        // 使用一个含有k个元素的最小堆
        PriorityQUeue<Integer> minHeap = new Priority<>(k,(a,b)->a-b);
        for(int i = 0; i<k;i++){
   
            minHeap.add(nums[i]);
        }
        
        for(int i = k; i<len;i++
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值