POJ 2823 Sliding Window 线段树区间求和问题

时间:2022-03-01 06:19:11

题目链接

线段树区间求和问题,维护一个最大值一个最小值即可,线段树要用C++交才能过。

注意这道题不是求三个数的最大值最小值,是求k个的。

本题数据量较大,不能用N建树,用n建树。

还有一种做法是单调队列,耗时更少。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define N 1000005
using namespace std;
int n,k;
struct Tree
{
int l,r,mx,mn;
}tree[N*];
void build(int root,int l,int r)
{
tree[root].l=l;
tree[root].r=r;
if(l==r)
{
scanf("%d",&tree[root].mx);
tree[root].mn=tree[root].mx;
return;
}
int mid=(l+r)>>;
build(root<<,l,mid);
build(root<<|,mid+,r);
tree[root].mx=max(tree[root<<].mx,tree[root<<|].mx);
tree[root].mn=min(tree[root<<].mn,tree[root<<|].mn);
}
int querymx(int root,int l,int r)
{
if(l<=tree[root].l&&r>=tree[root].r) return tree[root].mx;
int mid=(tree[root].l+tree[root].r)>>,ret=-;
if(l<=mid) ret=max(querymx(root<<,l,r),ret);
if(r>mid) ret=max(querymx(root<<|,l,r),ret);
return ret;
}
int querymn(int root,int l,int r)
{
if(l<=tree[root].l&&r>=tree[root].r) return tree[root].mn;
int mid=(tree[root].l+tree[root].r)>>,ret=;
if(l<=mid) ret=min(querymn(root<<,l,r),ret);
if(r>mid) ret=min(querymn(root<<|,l,r),ret);
return ret;
}
void pr()
{
for(int i=;i<=n-k+;i++)
printf("%d ",querymn(,i,i+k-));
puts("");
for(int i=;i<=n-k+;i++)
printf("%d ",querymx(,i,i+k-));
puts("");
}
int main()
{
while(scanf("%d%d",&n,&k)!=EOF)
{
build(,,n);
pr();
}
return ;
}