leetcode 703.数据流中的第K大元素

设计一个找到数据流中第K大元素的类（class）。注意是排序后的第K大元素，不是第K个不同的元素。

你的 KthLargest 类需要一个同时接收整数 k 和整数数组nums 的构造器，它包含数据流中的初始元素。每次调用 KthLargest.add，返回当前数据流中第K大的元素。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-largest-element-in-a-stream
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码如下

class KthLargest {
    
    int k ;
    PriorityQueue<Integer> queue ;
    
    public KthLargest(int k, int[] nums) //构造方法
    {
        this.k = k;
        queue = new PriorityQueue(k);
        Arrays.sort(nums);
        if(nums.length!=0&&nums.length>k)//数组不为空，且长度大于K，将后K个元素加入最小堆
        {for(int i=nums.length-1;i>nums.length-k-1;i--)
         {
            queue.add(nums[i]);
         }
        }
        else//否则，将所有元素加入最小堆；
        {
            for(int i:nums)
                queue.add(i);
        }
    }
    
    public int add(int val)
    {  
       if(queue.isEmpty()||queue.size()<k)//如果队列为空或队列长度小于K,直接将值加入最小堆
           queue.add(val);
       else if(val>queue.peek())//否则，如果值大于堆顶元素，则将堆顶元素移除，并将当前元素加入堆;
       {  
           queue.poll();
           queue.add(val);
           return queue.peek();
       }
        return queue.peek();
    }
}

以上通过优先队列最小堆构造类，完成功能。