Vijos 1083 小白逛公园 （线段树维护区间连续最大和）

题目链接：https://vijos.org/p/1083

解题思路：

线段树维护四个值：

l表示区间左到右最大值

r表示区间右到左最大值

sum表示区间和

ans表示区间连续最大和

这题感觉就是HDU 1540的升级版

http://acm.hdu.edu.cn/showproblem.php?pid=1540

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define mid int m=l+r>>1
#define lson rt<<1,l,m
#define rson rt<<1|1,m+1,r
#define tl tree[rt<<1]
#define tr tree[rt<<1|1]

using namespace std;

const int N = 5e5+5;

struct T
{
    int l,r,sum,ans;
}tree[N<<2];

void push_up(int rt)
{
    tree[rt].l = max(tl.l, tl.sum+tr.l);
    tree[rt].r = max(tr.r, tr.sum+tl.r);
    tree[rt].sum = tl.sum + tr.sum;
    tree[rt].ans = max(max(tl.ans,tr.ans),tl.r+tr.l);
}

void update(int p,int x,int rt,int l,int r)
{
    if (l==r){
        tree[rt].l = tree[rt].r = tree[rt].sum = tree[rt].ans = x;
        return ;
    }
    mid;
    if (p<=m) update(p,x,lson);
    else update(p,x,rson);
    push_up(rt);
}

T query(int L,int R,int rt,int l,int r)
{
    if (L==l && r==R){
        return tree[rt];
    }
    mid;
    if (L>m) return query(L,R,rson);
    else if (R<=m) return query(L,R,lson);
    T a,b,c;
    a = query(L,m,lson);
    b = query(m+1,R,rson);
    c.l = max(a.l,a.sum+b.l);
    c.r = max(b.r,b.sum+a.r);
    c.sum = a.sum + b.sum;
    c.ans = max(max(a.ans,b.ans),a.r+b.l);
    return c;
}

void build(int rt,int l,int r)
{
    if (l==r){
        scanf("%d",&tree[rt].ans);
        tree[rt].l = tree[rt].r = tree[rt].sum  = tree[rt].ans;
        return ;
    }
    mid;
    build(lson);
    build(rson);
    push_up(rt);
}

int main()
{
    int n,m;
    while (~scanf("%d %d",&n,&m)){
        build(1,1,n);
        int k,a,b;
        while (m--){
            scanf("%d %d %d",&k,&a,&b);
            if (k==1){
                if (a>b) swap(a,b);
                printf("%d\n",query(a,b,1,1,n).ans);
            }
            else update(a,b,1,1,n);
        }
    }
    return 0;
}

我学乖了，必不可能不多组输入。