首页 > 代码库 > AC日记——滑动窗口 洛谷 P1886
AC日记——滑动窗口 洛谷 P1886
题目描述
现在有一堆数字共N个数字(N<=10^6),以及一个大小为k的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:
The array is [1 3 -1 -3 5 3 6 7], and k = 3.
输入输出格式
输入格式:
输入一共有两行,第一行为n,k。
第二行为n个数(<INT_MAX).
输出格式:
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
输入输出样例
输入样例#1:
8 31 3 -1 -3 5 3 6 7
输出样例#1:
-1 -3 -3 -3 3 33 3 5 5 6 7
说明
50%的数据,n<=10^5
100%的数据,n<=10^6
思路:
裸线段树;
来,上代码:
#include <cstdio>#include <iostream>#include <algorithm>using namespace std;class T_tree { public: int l,r,dis_max,dis_min,mid; void dis_() { scanf("%d",&dis_max); dis_min=dis_max; } void mid_() { mid=(l+r)>>1; }};class T_tree tree[4000001];class T_ans { public: int minn,maxn;};class T_ans ans[1000001];int n,k;inline void tree_up(int now){ tree[now].dis_max=max(tree[now<<1].dis_max,tree[now<<1|1].dis_max); tree[now].dis_min=min(tree[now<<1].dis_min,tree[now<<1|1].dis_min);}void tree_build(int now,int l,int r){ tree[now].l=l,tree[now].r=r; if(l==r) { tree[now].dis_(); return ; } tree[now].mid_(); tree_build(now<<1,l,tree[now].mid); tree_build(now<<1|1,tree[now].mid+1,r); tree_up(now);}inline class T_ans ans_(int mi,int ma){ class T_ans cur_; cur_.maxn=ma; cur_.minn=mi; return cur_;}class T_ans tree_query(int now,int l,int r){ if(tree[now].l==l&&tree[now].r==r) { return ans_(tree[now].dis_min,tree[now].dis_max); } if(l>tree[now].mid) return tree_query(now<<1|1,l,r); else if(r<=tree[now].mid) return tree_query(now<<1,l,r); else { class T_ans A=tree_query(now<<1,l,tree[now].mid); class T_ans B=tree_query(now<<1|1,tree[now].mid+1,r); A.maxn=max(A.maxn,B.maxn); A.minn=min(A.minn,B.minn); return A; }}int main(){ scanf("%d%d",&n,&k); tree_build(1,1,n); for(int i=k;i<=n;i++) { ans[i-k+1]=tree_query(1,i-k+1,i); } for(int i=1;i<=n-k+1;i++) { printf("%d ",ans[i].minn); } printf("\n"); for(int i=1;i<=n-k+1;i++) { printf("%d ",ans[i].maxn); } return 0;}
AC日记——滑动窗口 洛谷 P1886
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。