Leetcode 703 数据流中的第K大元素

题目

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

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

示例:

int k = 3; int[] arr = [4,5,8,2]; KthLargest kthLargest = new
KthLargest(3, arr); kthLargest.add(3); // returns 4
kthLargest.add(5); // returns 5 kthLargest.add(10); // returns 5
kthLargest.add(9); // returns 8 kthLargest.add(4); // returns 8

说明: 你可以假设 nums 的长度≥ k-1 且k ≥ 1。

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

题解

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

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

	public KthLargest(int k, int[] nums) {
		this.k = k;
		this.pq = new PriorityQueue<Integer>(k);
		for (int i : nums) {
			add(i);
		}
	}

	public int add(int val) {

		if (pq.size() < k) {
			pq.offer(val);
		} else if (val > pq.peek()) {
			pq.poll();
			pq.offer(val);
		}
		return pq.peek();

	}
}

public class Leetcode703 {

	public static void main(String[] args) {
		int k = 3;
		int[] arr = { 4, 5, 8, 2 };
		KthLargest kthLargest = new KthLargest(k, arr);
		System.out.println(kthLargest.add(3));// returns 4
		System.out.println(kthLargest.add(5));// returns 5
		System.out.println(kthLargest.add(10));// returns 5
		System.out.println(kthLargest.add(9));// returns 8
		System.out.println(kthLargest.add(4));// returns 8
	}
}