紫书 滑动窗口最小值问题

看紫书看到了 滑动窗口最小值问题 觉得有必要来记录一下

首先是给出问题 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;
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值