CH4301 Can you answer on these queries III 题解

给定长度为N的数列A，以及M条指令 (N≤500000, M≤100000)，每条指令可能是以下两种之一：
“2 x y”，把 A[x] 改成 y。
“1 x y”，查询区间 [x,y] 中的最大连续子段和，即 max(x≤l≤r≤y)⁡ { (sum_{i=l}^r) A[i] }。
对于每个询问，输出一个整数表示答案。

一道模板题，线段树的区间最大子段和。

代码

#include<bits/stdc++.h>
using namespace std;
const int SIZE=500010;
struct node{
    long long data;
    long long sum;
    long long l,r;
    long long lmax,rmax;
}t[SIZE*4];
long long n,m,u,v,k,sum,a[SIZE];
void build(long long p,long long l,long long r){
    t[p].l=l,t[p].r=r;
    if(l==r){
        t[p].sum=t[p].data=t[p].lmax=t[p].rmax=a[l];
        return;
    } 
    long long mid=(l+r)/2;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    t[p].sum=t[p*2].sum+t[p*2+1].sum;
    t[p].lmax=max(t[p*2].lmax,t[p*2].sum+t[p*2+1].lmax);
    t[p].rmax=max(t[p*2+1].rmax,t[p*2+1].sum+t[p*2].rmax);
    t[p].data=max(max(t[p*2].data,t[p*2+1].data),t[p*2].rmax+t[p*2+1].lmax);
}
void change(long long p,long long x,long long v){
    if(t[p].l==t[p].r){
        t[p].data=t[p].sum=t[p].lmax=t[p].rmax=v;
        return;
    }
    long long mid=(t[p].l+t[p].r)/2;
    if(x<=mid) change(p*2,x,v);
    else change(p*2+1,x,v);
    t[p].sum=t[p*2].sum+t[p*2+1].sum;
    t[p].lmax=max(t[p*2].lmax,t[p*2].sum+t[p*2+1].lmax);
    t[p].rmax=max(t[p*2+1].rmax,t[p*2+1].sum+t[p*2].rmax);
    t[p].data=max(max(t[p*2].data,t[p*2+1].data),t[p*2].rmax+t[p*2+1].lmax);    
}
node ask(long long p,long long l,long long r){
    if(l<=t[p].l&&r>=t[p].r) return t[p];
    long long mid=(t[p].l+t[p].r)/2;
    if(mid>=r) return ask(p*2,l,r);
    if(mid<l) return ask(p*2+1,l,r);
    else{
        node ans,a1,b;
        a1=ask(p*2,l,r);
        b=ask(p*2+1,l,r);
        ans.sum=a1.sum+b.sum;
        ans.data=max(max(a1.data,a1.rmax+b.lmax),b.data);
        ans.lmax=max(a1.lmax,a1.sum+b.lmax);
        ans.rmax=max(b.rmax,b.sum+a1.rmax);
        return ans;
    }
}
int main(){
    scanf("%lld",&n);
    scanf("%lld",&m);
    for(int i=1;i<=n;++i){
        scanf("%lld",&a[i]);
    } 
    build(1,1,n);
    while(m--){
        scanf("%lld %lld %lld",&k,&u,&v);
        if(k==1){
            if(u>v) swap(u,v);
            printf("%lld
",ask(1,u,v).data);
        }
        else{
            change(1,u,v);
        }
    }
    return 0;
}