LC239 滑动窗口最大值
昨天尝试了暴力,然后学习了一种单调队列的方法。
首先定义一个单调队列,要求队列里能够按照从大到小来(从头到尾)排序。
所以在pop的时候要判断value==que.front()
,是的话pop_front()
,不是则跳过。
为了满足从大到小排列,在push的时候循环判断value>que.back()
,是的话要pop尾部的元素。最后统一push当前的值。
所以
class MyQueue{
public:
deque<int> que;
void push(int value){
while(!que.empty() && value>que.back()){
que.pop_back();
}
que.push_back(value);
}
void pop(int value){
if(!que.empty() && value==que.front()){
que.pop_front();
}
}
int front(){
return que.front();
}
};
然后对nums进行处理,先对前k个元素找到最大值,然后一个个pop头,push尾。最后输出结果。
class Solution {
private:
class MyQueue{
public:
deque<int> que;
void push(int value){
while(!que.empty() && value>que.back()){
que.pop_back();
}
que.push_back(value);
}
void pop(int value){
if(!que.empty() && value==que.front()){
que.pop_front();
}
}
int front(){
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que;
vector<int> ans;
for(int i=0;i<k;i++){
que.push(nums[i]);
}
ans.push_back(que.front());
for(int i=k;i<nums.size();i++){
que.pop(nums[i-k]);
que.push(nums[i]);
ans.push_back(que.front());
}
return ans;
}
};
LC347 前k个高频元素
首先用map来记录每个数和对应的频率。为了找到前k个高频元素,需要对map的内容进行排序。
可以用快排,这样的时间复杂度是O(nlogn)
。但是题目只用前k个,所以采用最大堆和最小堆方法。
如果使用最大堆的话,那为了维护k个元素,当弹出时会把最大的元素弹出去了,这就无法保留前k个高频元素。
如果采用最小堆方法,那么弹出的就是最小的元素,留下的就是最大的元素。
优先队列的内部实现就是用大顶堆和小顶堆来完成的,所以这里可以直接用优先队列来实现。
最后对小顶堆进行一个逆序输出就好了。
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int> umap;
for(auto x:nums){
umap[x]++;
}
//定义一个小顶堆
priority_queue<pair<int,int>,vector<pair<int,int>>,greater<>> pq;
//用固定大小为k的小顶堆来扫描umap
for(auto x:umap){
//按照频率排序,所以second放前面
pq.push({x.second,x.first});
if(pq.size()>k){
pq.pop();
}
}
//找到前k个元素,因为小顶堆先pop最小的,所以倒序来输出到数组
vector<int> ans(k);
for(int i=k-1;i>=0;i--){
ans[i]=pq.top().second;//前面把first和second换过顺序了,所以这里用second输出val的值
pq.pop();
}
return ans;
}
};