题意:让你构造一个类,初始化后,每次调用某成员函数可以得到历史数据的第K大的数(有两个相同的数算两个)。
解题思路:
思路一:初始化使用vector排序,每次数据来了重新排序,取下标第K个。提交,TLE;
思路二:初始化使用vector排序并取前K个数,每次新数据来了加入vector后再排序,取第K个。提交TLE。
思路三:初始化使用vector排序并取前K个数,每次新数据来了使用二分查找得到新元素应该在的位置,用vector.insert函数插入新数据,取第K个。提交AC,超过25%,很慢。
思路四:成员数据结构使用优先队列,并规定从小到大排序。初始化使用vector排序并取前K个数放入优先队列中,每次新数据来了,将新数据放入优先队列,如果队列长度大于K,剔除最上面的元素,返回优先队列最顶上的元素(最小),提交AC,超过98%。
class KthLargest {
private:
int kth;
priority_queue<int, vector<int>, greater<int> > q;
public:
KthLargest(int k, vector<int> nums) {
sort(nums.begin(), nums.end(), [](const int a, const int b){return a > b;});
int sizeN = nums.size();
for(int i = 0; i < k && i < sizeN; i++)
q.push(nums[i]);
kth = k;
}
int add(int val) {
q.push(val);
if(q.size() > kth)
q.pop();
return q.top();
}
};
static int x = [](){
std::ios::sync_with_stdio(false);
std::cin.tie(NULL);
return 0;
}();
/**
* Your KthLargest object will be instantiated and called as such:
* KthLargest obj = new KthLargest(k, nums);
* int param_1 = obj.add(val);
*/
注意:
C++优先队列的使用方法成员函数与普通队列几乎一样;
优先队列使用的其实是堆排序,大顶堆和小顶堆。
priority_queue<int> q1; //大顶堆。声明的时候默认是按照从大到小的顺序(大的在前面,先出队列)
priority_queue<int, vector<int>, greater<int> > q2 ; // 小顶堆。指定队列内装的元素类型(int),队列的容器(vector<int>), 队列的排序方式(从小到大排序,小的在队列前面,q.pop()和q.top()).
网友评论