数据流中的第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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

class KthLargest {
public:
    
    vector<int> tmp;
    int size;
    KthLargest(int k, vector<int>& nums) 
    {
        if(nums.size() > k)
        {
            sort(nums.rbegin(), nums.rend());
            tmp = vector<int>(nums.begin(), nums.begin() + k);
            // for(int i = 0; i < tmp.size(); i++)
            //     cout<<tmp[i]<<" "<<endl;
        }
        else if(nums.size() == k)
        {
            sort(nums.rbegin(), nums.rend());
            tmp = nums;
        }
        else
        {
            tmp = nums;
            sort(tmp.rbegin(), tmp.rend());
            for(int i = 0; i < tmp.size(); i++)
                cout<<tmp[i]<<" "<<endl;
        }
        
        size = k;
    }
    
    int add(int val) 
    {
        if(tmp.size() < size)
        {
            tmp.push_back(val);
            sort(tmp.rbegin(), tmp.rend());
            return tmp[tmp.size() - 1];
        }
        else if(tmp.size() == size)
        {
            if(val <= tmp[size - 1])
                return tmp[size - 1];
            else
            {
                // tmp[size-1] = val;
                // sort(tmp.rbegin(), tmp.rend());
                // return tmp[size-1];
                int i = tmp.size() - 1;
                for(; i >= 0; i--)
                {
                    if(tmp[i] > val)
                    {
                        tmp.insert(tmp.begin()+i+1,val);
                        tmp.erase(tmp.begin() + tmp.size() - 1);
                        //cout<<tmp.size()<<endl;
                        break;
                    }
                }
                if(i < 0)
                {
                    tmp.insert(tmp.begin(),val);
                    tmp.erase(tmp.begin() + tmp.size() - 1);
                }
                
                return tmp[tmp.size()-1];
                
                
            }
        }
        return 0;
    }
};

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