莫欺少年穷,修魔之旅在这开始—>算法提高课题解
思路:
1. 先建立线段树(注意:回溯时不要忘了更新)
2. 修改操作:直接修改 tr[u].l == x && tr[u].r == x 位置的数(回溯时不要忘了更新)
3. 查询操作则有以下四种情况:
4. (1)若该树区间完全包含在被查询的区间内,则直接返回该树区间
5. (2)若查询区间在 mid 左边,则递归左边
6. (3)若查询区间在 mid 右边,则递归右边
7. (4)若查询区间跨越 mid,则两边都递归并 pushup 一遍
#include<bits/stdc++.h>
using namespace std;
const int N = 500010;
int n,m;
int a[N];
struct Node
{
int l,r;
int sum,lmax,rmax,tmax;
}tr[N*4]; //开四倍空间
void pushup(Node &u,Node &l,Node &r)
{
//区间总和
u.sum=l.sum+r.sum;
//最大前缀和
u.lmax=max(l.lmax,l.sum+r.lmax);
//最大后缀和
u.rmax=max(r.rmax,r.sum+l.rmax);
//最大连续子段和
u.tmax=max(max(l.tmax,r.tmax),l.rmax+r.lmax);
}
//由子节点求父节点的最大连续子段和
void pushup(int u)
{
pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}
void build(int u,int l,int r)
{
if(l==r) tr[u]={l,r,a[l],a[l],a[l],a[l]};
else
{
tr[u]={l,r};
int mid=l+r>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
//回溯不要忘了更新
pushup(u);
}
}
void modify(int u,int x,int v)
{
if(tr[u].l==tr[u].r&&tr[u].l==x) tr[u]={x,x,v,v,v,v};
else
{
int mid=tr[u].l+tr[u].r>>1;
if(x<=mid) modify(u<<1,x,v);
else modify(u<<1|1,x,v);
//回溯不要忘了更新
pushup(u);
}
}
Node query(int u,int l,int r)
{
if(tr[u].l>=l&&tr[u].r<=r) return tr[u];
else
{
int mid=tr[u].l+tr[u].r>>1;
//该区间全部在 mid 左边
if(r<=mid) return query(u<<1,l,r);
//该区间全部在 mid 右边
else if(l>mid) return query(u<<1|1,l,r);
//该区间跨越 mid
else
{
auto left=query(u<<1,l,r);
auto right=query(u<<1|1,l,r);
Node res;
pushup(res,left,right);
return res;
}
}
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];
//建立线段树
build(1,1,n);
while(m--)
{
int k,x,y;
cin>>k>>x>>y;
//查询操作
if(k==1)
{
if(x>y) swap(x,y);
cout<<query(1,x,y).tmax<<endl;
}
//修改操作
else modify(1,x,y);
}
return 0;
}
666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666666