如何使用大顶堆和小顶堆找topN、如何从大量数据中找出高频词?

如何使用大顶堆和小顶堆找topN、如何从大量数据中找出高频词?,第1张

如何使用大顶堆和小顶堆找topN
  • 我们如何从一个链表或者数组中,找到第k大的数,或者前k大的数。使用小顶堆。(输出是从小到大排列的前K大的数
  • 我们如何从一个链表或者数组中,找到第k小的数,或者前k小的数(输出的堆是堆顶最大,其他不是序列顺序),使用大顶堆。
    小顶堆找TopN元素:
一、使用java.util.PriorityQueue优先级队列,默认是从小到大排序,初始指定最大容量为N 二、当size()大小=N的时候将新val与堆顶最小的比较替换(其中queue.poll是d出数组第一个元素,也即是堆顶的最小元素元素)

可运行完整代码:

package com.排序;
import java.util.PriorityQueue;

public class 小顶堆找最大TopN {
    static int N = 4;
    static PriorityQueue<Integer> queue;   // 优先级队列默认小顶堆
    public static void main(String[] args) {
        queue = new PriorityQueue<Integer>(N);
        int[] arr = {1, 3, 5, 2, 6};
        for (int num : arr) {   // 把数组元素放到堆中
            小顶堆找TopN.add(num);
        }
    }

    private static void add(int val) {   // 选出前N大的数
        if (queue.size() < N) {   // 先比较,再添加(size=3,添加元素2,之后此if走完size就会变为4)
            queue.offer(val);
        } else if (queue.size() >= N && queue.peek() < val) {   // 小顶堆堆顶存放堆中最小的数,如果有val比堆顶元素大,就d出最小(堆顶) 并 添加新元素
            queue.poll();
            queue.offer(val);
        }
        System.out.println("数组中第" + N + "大的数是:" + queue.peek());
        System.out.println("从小到大前N个数的堆为:" + queue);
    }
}
如何从大量数据中找出高频词?

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/735552.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-27
下一篇 2022-04-27

发表评论

登录后才能评论

评论列表(0条)

保存