LeetCode 703题数据流中的第K大元素（Kth Largest Element in a Stream）

import java.util.PriorityQueue;

class KthLargest {

    private PriorityQueue<Integer> pri_queue;
    private int k;
    public KthLargest(int k, int[] nums) {
        //将k赋值给类的全局变量
        this.k = k;
        //创建一个大小为k的优先队列
        pri_queue = new PriorityQueue<>(k);
        for(int val : nums){
            add(val);
        }
    }

    public int add(int val) {
        //如果pri_queue没有满
        if(pri_queue.size()<k){
            //直接将val加入队列，并调整
            pri_queue.offer(val);
        }
        //如果堆顶元素小于val
        else if(pri_queue.peek() < val){
            //弹出堆顶元素
            pri_queue.poll();
            //加入val并调整
            pri_queue.offer(val);
        }

        //返回堆顶元素
        return pri_queue.peek();
    }
}

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest obj = new KthLargest(k, nums);
 * int param_1 = obj.add(val);
 */