239. 滑动窗口最大值(题目链接:力扣)
思路:想到要用队列做,也知道队列里面需要排序,但是不知道每次移出来的元素怎么处理。看解析知道了原来是要比较队列出口元素和移除元素是否相等,相等的话就移出来,否则还让其呆在队列里面充当最大元素的角色,值得注意的是这题需要用到双端队列做,也就是deque<int>,头文件中需要引入#include<deque>,每次push进来元素的时候要将该元素和队尾的元素意义比较,把比自身小的元素全部从队列中pop_back()出来。
class MyQue{
public:
deque<int> que;
void pop(int value){
if(value == que.front()){
que.pop_front();
}
}
void push(int value){
while(!que.empty() && que.back() < value){
que.pop_back();
}
que.push_back(value);
}
int front(){
return que.front();
}
};
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQue que;
vector<int> result;
int size = nums.size();
for(int i=0; i<k; i++){
que.push(nums[i]);
}
result.push_back(que.front());
for(int i=k; i<size; i++){
que.push(nums[i]);
que.pop(nums[i-k]);
result.push_back(que.front());
}
return result;
}
347.前 K 个高频元素(题目链接:力扣)
思路:元素出现次数显然要用到map<int, int>来存储,然后又是排序,能想到要用队列,但是不知道队列里面如何处理才能时间复杂度小于nlogn,看完解析才知道用优先级队列priority_queue(底层为完全二叉树),将map中的pair<int,int>push进来的同时就在排序,按理来说用大根堆或者小根堆都可,但是大根堆的排法今晚始终都没调出来(这里挖个坑,之后必定深入研究理解)。这种方法的时间复杂度为nlogk,属于用空间换时间的典型做法,空间复杂度n.
class recomparison{
public:
bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs){
return lhs.second > rhs.second;
}
};
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> map;
for(int i=0; i<nums.size(); i++){
map[nums[i]]++;
}
priority_queue<pair<int, int>, vector<pair<int, int>>, recomparison> pri_que;
for(unordered_map<int, int>::iterator it=map.begin(); it!=map.end(); it++){
pri_que.push(*it);
if(pri_que.size() > k){
pri_que.pop();
}
}
vector<int> result(k);
for(int i=k-1; i>=0; i--){
result[i] = pri_que.top().first;
pri_que.pop();
}
return result;
}