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 3
1 3 -1 -3 5 3 6 7

输出样例#1： 

-1 -3 -3 -3 3 3
3 3 5 5 6 7

说明

50%的数据，n<=10^5

100%的数据，n<=10^6

 

Solution：

　　本题是单调队列的模板题。。。

　　单调队列思想虽简单，但是代码各种判断很容易出错，于是打了一个线段树水一波分。

　　思路就是区间查询最值，维护一下就$OK$了。

代码：

 

#include<bits/stdc++.h>
#define il inline
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int inf=2147483647,N=1e6+7;
using namespace std;
int n,k,cnt,ans1[N],ans2[N];
struct node{
    int maxn,minn;
}tr[N<<2];
il int gi(){
    int a=0;char x=getchar();bool f=0;
    while((x<'0'||x>'9')&&x!='-')x=getchar();
    if(x=='-')x=getchar(),f=1;
    while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
    return f?-a:a;
}
il int Max(int a,int b){return a>b?a:b;}
il int Min(int a,int b){return a>b?b:a;}
il void pushup(int rt){
    tr[rt].maxn=Max(tr[rt<<1].maxn,tr[rt<<1|1].maxn);
    tr[rt].minn=Min(tr[rt<<1].minn,tr[rt<<1|1].minn);
}
il void build(int l,int r,int rt){
    if(l==r){tr[rt].maxn=tr[rt].minn=gi();return;}
    int m=l+r>>1;
    build(lson),build(rson);
    pushup(rt);
}
il void query(int L,int R,int l,int r,int rt,int cnt){
    if(L<=l&&R>=r){ans1[cnt]=Min(ans1[cnt],tr[rt].minn);ans2[cnt]=Max(ans2[cnt],tr[rt].maxn);return;}
    int m=l+r>>1;
    if(L<=m)query(L,R,lson,cnt);
    if(R>m)query(L,R,rson,cnt);
}
int main(){
    n=gi(),k=gi()-1;
    for(int i=1;i<=n;i++)ans1[i]=inf,ans2[i]=-inf;
    build(1,n,1);
    for(int i=1;i+k<=n;i++)query(i,i+k,1,n,1,++cnt);
    for(int i=1;i<=cnt;i++)printf("%d ",ans1[i]);
    printf("\n");
    for(int i=1;i<=cnt;i++)printf("%d ",ans2[i]);
    return 0;
}