题目传送门
题目大意
有一个长为 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;
}