海量数据topK问题

给你一亿个数据,从中找出前K个大的数据。
有两种解决办法。

1.直接将数据从大到小排序,然后取前K个。但是由于数据的数量过于庞大,要开辟很大的空间,很浪费内存,所以这种方法不建议使用。

2.用堆来解决。要找前K个大的数据,则将待找的元素的前K个元素建立大小为K的小根堆,小根堆的堆顶元素是这K个数据中最小的。从剩下的数据里取一个元素每次和堆顶元素比较,如果数组当中的当前元素大于堆顶元素,则将堆顶元素出堆,然后将当前元素入堆。再将其调整为小堆,如此再循环。
代码如下:

    public static Integer[] findKNum(int[] array,int k){
        //1.建立大小为K的堆
        PriorityQueue<Integer> minHeap = new PriorityQueue<>
                (k, new Comparator<Integer>() {
                    //调整的时候使用
            @Override
            public int compare(Integer o1, Integer o2) {
                return o1.compareTo(o2);//建立小堆
                //return o2.compareTo(o1);建立大堆
            }
        });
        //2.
        for(int i=0; i<array.length; i++){
            if(minHeap.size()<k){
                minHeap.add(array[i]);//前K个元素入堆
            }else{
                Integer top = minHeap.peek();
                if(top!=null && top<array[i]){
                    minHeap.poll();
                    minHeap.add(array[i]);
                }
            }
        }
        Integer[] integers = new Integer[k];
        for(int i=0; i<k; i++){
            Integer top = minHeap.peek();
            integers[i] = top;//从小到大输出
            minHeap.poll();
        }
        return  integers;
    }
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值