[LeetCode] 703. Kth Largest Element in a Stream - Java

• 문제 링크

Kth Largest Element in a Stream - LeetCode

 

• 풀이 코드

import java.util.PriorityQueue;

public class KthLargest {

    private final PriorityQueue<Integer> pq;
    private final int k;

    public KthLargest(int k, int[] nums) {
        this.k = k;
        this.pq = new PriorityQueue<>(k);

        for (int num : nums) add(num);
    }

    public int add(int val) {
        pq.offer(val);
        if (pq.size() > k) pq.poll();

        return pq.element();
    }

}