这道题难点在于如何在logn的范围内找到该区间最大值,我们不妨先分析最大值有哪几种情况。
如图所示:
可分为3种情况
1.全部在中线左边。
2.全部在中线右边。
3.经过跨过中线。
如果在二叉树上分析,全部在中线左边即为左儿子的最大值,全部在中线右边即为右儿子的最大值,跨国中线的情况需要分析左儿子包含右端点的最大值和右儿子包含左端点的最大值。
如图所示:
代码实现如下
void update(int k)
{
tr[k].v=tr[k*2].v+tr[k*2+1].v;
tr[k].lm=max(tr[k*2].lm,tr[k*2].v+tr[k*2+1].lm);
tr[k].rm=max(tr[k*2+1].rm,tr[k*2+1].v+tr[k*2].rm);
tr[k].mm=max(tr[k*2].mm,max(tr[k*2+1].mm,tr[2*k].rm+tr[k*2+1].lm));
}
注意由于搜索时可能会遇到过中线的情况所以需要进行和update一样的操作,我们直接声明一个结构体函数;
代码如下:
node sear(int l,int r,int k)//由于和update有一样操作直接找结构体
{
node left1,right1;
int f1=0,f2=0;
if(tr[k].l>=l && tr[k].r<=r)
return tr[k];
int mid=(tr[k].l+tr[k].r)/2;
if(l<=mid)
left1=sear(l,r,k*2),f1=1;//如果区间包含中线左边 。
if(r>mid)
right1=sear(l,r,k*2+1),f2=1;//如果区间包含中线右边。
if(f2&&f1)//如果区间同时包含中线左右边则需进行update时一样的操作。
{
node m;
m.l=left1.l,m.r=right1.r,m.v=left1.v+right1.v;
m.lm=max(left1.lm,left1.v+right1.lm);
m.rm=max(right1.rm,right1.v+left1.rm);
m.mm=max(left1.mm,max(right1.mm,left1.rm+right1.lm)) ;
return m;
}
if(f1)//只包含左边;
return left1;
if(f2)//只包含右边;
return right1;
}
接着便是线段数的常见操作:
完整代码如下:
#include<cstdio>
#include<iostream>
using namespace std;
struct node
{
int l,r,mm,lm,rm,v;
}tr[2000010];
void update(int k)
{
tr[k].v=tr[k*2].v+tr[k*2+1].v;
tr[k].lm=max(tr[k*2].lm,tr[k*2].v+tr[k*2+1].lm);
tr[k].rm=max(tr[k*2+1].rm,tr[k*2+1].v+tr[k*2].rm);
tr[k].mm=max(tr[k*2].mm,max(tr[k*2+1].mm,tr[2*k].rm+tr[k*2+1].lm));
}
void build(int l,int r,int k)
{
tr[k].l=l;
tr[k].r=r;
if(l==r)
{
cin>>tr[k].v;
tr[k].lm=tr[k].v;
tr[k].rm=tr[k].v;
tr[k].mm=tr[k].v;
return;
}
int mid=(l+r)/2;
build(l,mid,k*2);
build(mid+1,r,k*2+1);
update(k);//回溯时求该区间最大值。
}
void change(int pos,int num,int k)
{
if(tr[k].l==pos && tr[k].r==pos)
{
tr[k].v=num;
tr[k].lm=num,tr[k].rm=num,tr[k].mm=num;
return;
}
int mid=(tr[k].l+tr[k].r)>>1;
if(pos<=mid)
change(pos,num,k*2);
else
change(pos,num,k*2+1);//向下二分搜索 ;
update(k);//回溯时改变区间最大值;
}
node sear(int l,int r,int k)//由于和update有一样操作直接找结构体
{
node left1,right1;
int f1=0,f2=0;
if(tr[k].l>=l && tr[k].r<=r)
return tr[k];
int mid=(tr[k].l+tr[k].r)/2;
if(l<=mid)
left1=sear(l,r,k*2),f1=1;//如果区间包含中线左边 。
if(r>mid)
right1=sear(l,r,k*2+1),f2=1;//如果区间包含中线右边。
if(f2&&f1)//如果区间同时包含中线左右边则需进行update时一样的操作。
{
node m;
m.l=left1.l,m.r=right1.r,m.v=left1.v+right1.v;
m.lm=max(left1.lm,left1.v+right1.lm);
m.rm=max(right1.rm,right1.v+left1.rm);
m.mm=max(left1.mm,max(right1.mm,left1.rm+right1.lm)) ;
return m;
}
if(f1)//只包含左边;
return left1;
if(f2)//只包含右边;
return right1;
}
int main()
{
int n,m,type,le,ri;
cin>>n>>m;
build(1,n,1);
for(int i=1;i<=m;i++)
{
cin>>type>>le>>ri;
if(type==1 && le>ri)
swap(le,ri);
if(type==1)
{
node ans=sear(le,ri,1);
cout<<ans.mm<<endl;
}
else
change(le,ri,1);
}
return 0;
}