编程语言
首页 > 编程语言> > 算法——得到数据流中前K大的数

算法——得到数据流中前K大的数

作者:互联网

用优先队列

    public PriorityQueue<Integer> kthLargest(int k, int[]a) {
        PriorityQueue<Integer> q = new PriorityQueue<>(k);
        for (int i : a) {
            if (q.size() < k) {
                q.offer(i);
            }else {
                if (i > q.peek()) {
                    q.poll();
                    q.offer(i);
                }
            }
        }
        return q;
    }

 

标签:peek,return,offer,中前,PriorityQueue,int,算法,数据流
来源: https://www.cnblogs.com/gaoquanquan/p/10848261.html