Class KthLargest

KthLargestElementInAStream.java:1–21

class KthLargest {
    PriorityQueue<Integer> pq;
    int k;
    public KthLargest(int k, int[] nums) {
        this.k = k;
        pq = new PriorityQueue<>();
        for(int num : nums){
            add(num);
        }
    }
    
    public int add(int val) {
        if(pq.size()<k || val > pq.peek()){
            pq.offer(val);
            if(pq.size()>k){
                pq.poll(); 
            }
        }
        return pq.peek();
    }
}

/**
 * Your KthLargest object will be instantiated and called as such: