CH4301 Can youanswer on these queries III （线段树）

4301 Can you answer on these queries III 0x40「数据结构进阶」例题

描述

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

输入格式

第一行两个整数N,M

第二行N个整数Ai

接下来M行每行3个整数k,x,y，k=1表示查询（此时如果x>y，请交换x,y），k=2表示修改

输出格式

对于每个询问输出一个整数表示答案。

样例输入

5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 3 2

样例输出

2
-1

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
struct node{
    int l,r;
    int dat;
    int sum;
    int lmax;
    int rmax;
    node(){}
    node(int ll,int rr,int v){l=ll;r=rr;dat=sum=lmax=rmax=v;sum=0;}
} t[10*500005];
int n;
const int inf=0x3f3f3f3f;
int a[500005];
void build(int id,int L,int R)
{
    t[id].l=L;t[id].r=R;
    if(L==R){t[id].dat=a[L];t[id].sum=a[L];t[id].lmax=t[id].rmax=a[L];return;}
    int mid=(L+R)/2;
    build(id*2,L,mid);
    build(id*2+1,mid+1,R);
    t[id].sum=t[id*2].sum+t[id*2+1].sum;
    t[id].lmax=max(t[id*2].lmax,t[id*2].sum+t[id*2+1].lmax);
    t[id].rmax=max(t[id*2+1].rmax,t[id*2+1].sum+t[id*2].rmax);
    t[id].dat=max(t[id*2].dat,(max(t[id*2+1].dat,t[id*2].rmax+t[id*2+1].lmax)));
}
void change(int id,int pos,int v)
{
    int L=t[id].l;int R=t[id].r;
    if(L==R){t[id].sum=t[id].lmax=t[id].dat=t[id].rmax=v;return ;}
    int mid=(L+R)/2;
    if(pos<=mid)change(id*2,pos,v);
    else change(id*2+1,pos,v);
    t[id].sum=t[id*2].sum+t[id*2+1].sum;
    t[id].lmax=max(t[id*2].lmax,t[id*2].sum+t[id*2+1].lmax);
    t[id].rmax=max(t[id*2+1].rmax,t[id*2+1].sum+t[id*2].rmax);
    t[id].dat=max(t[id*2].dat,(max(t[id*2+1].dat,t[id*2].rmax+t[id*2+1].lmax)));
}
node ask(int id,int l,int r)
{
    int L=t[id].l;int R=t[id].r;
    if(l<=L&&r>=R)
    {
        node ans(L,R,t[id].dat);
        ans.lmax=t[id].lmax;ans.rmax=t[id].rmax;
        ans.sum=t[id].sum;
        return ans;
    }
    int mid=(L+R)/2;
    node left(L,mid,-inf);
    node right(mid+1,R,-inf);
   
   if(mid>=l) 
    left=ask(id*2,l,r);
    if(mid+1<=r) 
   right=ask(id*2+1,l,r);
    node res(L,R,-inf);
    int data;int lmax,rmax,sum;
    sum=left.sum+right.sum;
    lmax=max(left.lmax,left.sum+right.lmax);
    rmax=max(right.rmax,right.sum+left.rmax);
    data=max(left.dat,max(right.dat,left.rmax+right.lmax));
    res.dat=data;res.sum=sum;res.lmax=lmax;res.rmax=rmax;
    return res;
}
int q;
int main()
{
    ios::sync_with_stdio(0);
    cin>>n>>q;
    for(int i=1;i<=n;i++)cin>>a[i];
    build(1,1,n);
    int ll,rr;int ff;
    while(q--)
    {
        cin>>ff>>ll>>rr;
        
        if(ff==2)
        {
            change(1,ll,rr);
        }
        else if(ff==1)
        {
            if(ll>rr)swap(ll,rr);
            cout<<ask(1,ll,rr).dat<<endl;
        }
    }
    return 0;
}