线段树处理最大连续子段和问题

题目描述

#include <bits/stdc++.h>

using namespace std;
#define ll long long
const int maxn=5e5+10;
int m,n,ans,p;
int w[maxn];
struct node
{
    int l,r;
    int sum,lmax,rmax,tmax;

} tr[maxn*4];

//重写pushup函数
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,w[r],w[r],w[r],w[r]};
    else
    {
        tr[u]= {l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(u);
    }
}

int modify(int u,int x,int v)
{
    if(tr[u].l==x&&tr[u].r==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];
    int mid=tr[u].l+tr[u].r>>1;
    if(r<=mid)return query(u<<1,l,r);
    else if(l>mid)return query(u<<1|1,l,r);
    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()
{

    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&w[i]);
    }
    build(1,1,n);
    int k,x,y;
    while(m--)
    {
        scanf("%d%d%d",&k,&x,&y);
        if(k==1)
        {
            if(x>y)swap(x,y);
            printf("%d\n",query(1,x,y).tmax);
        }
        else modify(1,x,y);
    }

    return 0;
}