LeetCode 703. 数据流中的第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。

class KthLargest {
	int K;
	priority_queue<int,vector<int>,greater<int>> q;//小顶堆
public:
    KthLargest(int k, vector<int>& nums) {
        K = k;
        for(int i:nums)
        	add(i);
    }
    
    int add(int val) {
        if(q.size() < K)//没满
        	q.push(val);
        else//满了
        {
        	if(val > q.top())//要存最大的K个，比堆顶大，换掉它
        	{
        		q.pop();
        		q.push(val);
        	}
        }
        return q.top();
    }
};