首页 > 代码库 > 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