P1886 滑动窗口 /【模板】单调队列(线段树做法)

发布于 2020-10-06  0 次阅读


题目大意:

P1886 滑动窗口 /【模板】单调队列
有一个长为 n 的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。

思路:

本来是单调队列的题,因为练习线段树,所以用线段树写了一遍。
维护区间最大值和最小值即可,注意一下序列a的数据范围就行

AC Code:

#include<bits/stdc++.h>
using namespace std;
#define INF 0x7fffffff
#define int long long
const int N=1e6+9;
struct node{
    int score;
}a[N];
struct segtree{
    int maxv, minv;
}tr[N<<2];
inline int read()
{
    int ans=0;
    char last=' ',ch=getchar();
    while(ch<'0'||ch>'9') last=ch,ch=getchar();
    while(ch>='0'&&ch<='9') ans=ans*10+ch-'0',ch=getchar();
    if(last=='-') ans=-ans;
    return ans;
}
inline int lc(int p)   {return p<<1;}
inline int rc(int p)   {return p<<1|1;}
inline void build(int p, int l, int r){
    if(l==r){
        tr[p].maxv=tr[p].minv=a[l].score;
        return ;
    }
    int mid=(l+r)>>1;
    build(lc(p), l, mid);
    build(rc(p), mid+1, r);
    tr[p].maxv=max(tr[lc(p)].maxv, tr[rc(p)].maxv);
    tr[p].minv=min(tr[lc(p)].minv, tr[rc(p)].minv);
}
inline int query1(int p, int l, int r, int x, int y){
    if(x>r || y<l)  return -INF;
    if(l<=x && y<=r) return tr[p].maxv;
    int mid=(x+y)>>1;
    return max(query1(lc(p), l, r, x, mid), query1(rc(p), l, r, mid+1, y));
}
inline int query2(int p, int l, int r, int x, int y){
    if(x>r || y<l)  return INF;
    if(l<=x && y<=r) return tr[p].minv;
    int mid=(x+y)>>1;
    return min(query2(lc(p), l, r, x, mid), query2(rc(p), l, r, mid+1, y));
}
int n, m;
int mx[N], mi[N];

signed main(){
    n=read();
    m=read();
    for(int i=1; i<=n; i++){
        a[i].score=read();
    }
    build(1,1,N);
    for(int i=1; i<=n-m+1; i++)
        cout<<query2(1,i,i+m-1,1,N)<<" ";
    cout<<endl;
    for(int i=1; i<=n-m+1; i++)
        cout<<query1(1,i,i+m-1,1,N)<<" ";
    cout<<endl;
    return 0;
}

平平无奇的在校大学生