看紫书看到了 滑动窗口最小值问题 觉得有必要来记录一下
首先是给出问题 5,2,6,8,10,7,4 k=4
使用优先队列:
#include<iostream>
#include<queue>
#include<vector>
using namespace std;
int a[1000005];
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
vector<int> v_min;
vector<int> v_max;
for(int k=1;k<=n;k++){
int start=1;
int last=k;
while(last<=n){
priority_queue<int> pq1;//大顶堆
priority_queue<int,vector<int>,greater<int> > pq2;//小顶堆
for(int i=start;i<=last;i++) pq2.push(a[i]);
for(int i=start;i<=last;i++) pq1.push(a[i]);
v_min.push_back(pq2.top());
v_max.push_back(pq1.top());
start++;
last++;
//cout<<"last="<<last<<endl;
}
cout<<"k="<<k<<endl;
for(int i=0;i<v_min.size();i++) cout<<v_min[i]<<" ";
cout<<endl;
for(int i=0;i<v_max.size();i++) cout<<v_max[i]<<" ";
cout<<endl;
v_min.clear();
v_max.clear();
}
}
/*
8
1 3 -1 -3 5 3 6 7
*/
这个代码求的是k=1-n情况下,序列的最大值和最小值.
单调队列
#include<iostream>
#include<vector>
using namespace std;
int a[1000];
int que[1000];
vector<int> v1;
int main(){
int n,k;
cin>>n>>k;
for(int i=1;i<=n;i++) cin>>a[i];
int cur=1;
int front=0,rear=0;
que[rear++]=a[cur];
cur++;
for(;cur<=k;cur++){//将前k个加入 并删除无用元素
while(front<rear&&a[cur]<que[rear-1]) rear--;
que[rear++]=a[cur];
}
for(int i=front;i<rear;i++) cout<<que[i]<<" ";
cout<<endl;
v1.push_back(que[front]);
for(;cur<=n;cur++){
if(a[cur-k]==que[front]) front++;//s首先删除滑动前窗口的最左边元素(如果是有用元素的话)
while(front<rear&&a[cur]<que[rear-1]){//比新元素大的元素都变为无用 从右往左删除
rear--;
}
//cout<<a[cur];
que[rear++]=a[cur];
for(int i=front;i<rear;i++) cout<<que[i]<<" ";
cout<<endl;
v1.push_back(que[front]);
}
for(int i=0;i<v1.size();i++) cout<<v1[i]<<" ";
}
/*
7 4
5 2 6 8 10 7 4
*/
注释已经很清楚了…刚学习单调队列有点懵 重要的还是跟着代码一步步走,才能理解为什么叫做单调队列。
最后是 poj2823一道练习 单调队列模板题
不知道为什么超5000MS还能过。。
一开始用cin cout一直过不了 害
#include<iostream>
#include<vector>
using namespace std;
const int maxn=1000005;
int a[maxn];
int que_min[maxn];//单调递增队列
int que_max[maxn];//单调递减队列
int f_min=0,r_min=0;
int f_max=0,r_max=0;
vector<int> v_min;
vector<int> v_max;
int main(){
int n,k;
scanf("%d %d",&n,&k);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
int cur=1;
if(k==1)
{
for(int i=1;i<=n;i++)
cout<<a[i]<<' ';cout<<endl;
for(int i=1;i<=n;i++)
cout<<a[i]<<' ';cout<<endl;
return 0;
}
que_min[r_min++]=a[cur];
que_max[r_max++]=a[cur];
cur++;
//先加入前k个
for(;cur<=k;cur++){
while(f_min<r_min&&que_min[r_min-1]>a[cur]) r_min--;//删除无用元素
que_min[r_min++]=a[cur];
while(f_max<r_max&&que_max[r_max-1]<a[cur]) r_max--;//删除无用元素
que_max[r_max++]=a[cur];
}
v_min.push_back(que_min[f_min]);
v_max.push_back(que_max[f_max]);
for(;cur<=n;cur++){
if(a[cur-k]==que_min[f_min]) f_min++;
if(a[cur-k]==que_max[f_max]) f_max++;//先删除再加
while(f_min<r_min&&que_min[r_min-1]>a[cur]) r_min--;
que_min[r_min++]=a[cur];
while(f_max<r_max&&que_max[r_max-1]<a[cur]) r_max--;
que_max[r_max++]=a[cur];
v_min.push_back(que_min[f_min]);
v_max.push_back(que_max[f_max]);
}
for(int i=0;i<v_min.size();i++) printf("%d ",v_min[i]);
cout<<endl;
for(int i=0;i<v_max.size();i++) printf("%d ",v_max[i]);
cout<<endl;
}
LeetCode 239 滑动窗口最大值问题
我刚开始写的优先队列有点笨~做了每日一题看到了官方解法才恍然大悟
class Solution {
public:
typedef pair<int,int> P;
struct cmp{
bool operator()(const P a,const P b){
return a.first<b.first;//按照nums[i]递减排序
}
};
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int len=nums.size();
priority_queue<P,vector<P>,cmp> pq;
vector<int> res;
//初始时,我们将数组 nums 的前 k 个元素放入优先队列中。每当我们向右移动窗口时,我们就可以把一个新的元素放入优先队列中,此时堆顶的元素就是堆中所有元素的最大值
//然而这个最大值可能并不在滑动窗口中,在这种情况下,这个值在数组 nums 中的位置出现在滑动窗口左边界的左侧。因此,当我们后续继续向右移动窗口时,这个值就永远不可能出现在滑动窗口中了,我们可以将其永久地从优先队列中移除。
//我们不断地移除堆顶的元素,直到其确实出现在滑动窗口中。此时,堆顶元素就是滑动窗口中的最大值。
//时间复杂度为O((n-k)*logk)
for(int i=0;i<k;i++){
pq.push(make_pair(nums[i],i));
}
res.push_back(pq.top().first);
for(int i=k;i<len;i++){
pq.push(make_pair(nums[i],i));
while(pq.top().second<=i-k) pq.pop();
res.push_back(pq.top().first);
}
return res;
}
};
单调队列写法,我觉得比紫书的更好理解点吧
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> dq;//使用一个队列存储所有还没有被移除的下标。在队列中,这些下标按照从小到大的顺序被存储,并且它们在数组 nums 中对应的值是严格单调递减的
vector<int> res;
int len=nums.size();
for(int i=0;i<k;i++){
while(!dq.empty()&&nums[i]>=nums[dq.back()]) dq.pop_back();
dq.push_back(i);
}
res.push_back(nums[dq.front()]);
for(int i=k;i<len;i++){
//cout<<i<<endl;
//窗口内所有比当前值小的都变成无用的 移除
while(!dq.empty()&&nums[i]>=nums[dq.back()]) dq.pop_back();
//不在窗口内的需要移除
while(!dq.empty()&&dq.front()<=i-k) dq.pop_front();
dq.push_back(i);
res.push_back(nums[dq.front()]);
}
return res;
}
};