线段树-区间最大连续和 NKOJ1316 (小白逛公园)

#include <stdio.h>
#include <algorithm>

using namespace std;

const int _N = 550000;
const int INF = 1e9;

int A[_N];
int n, m;

struct node {
    int l, r, c_l, c_r, max, sum, mid;
    void set_lr(int _l, int _r) { l = _l; r = _r; mid = l+r>>1; return; }
    void set_val(int _v) { c_l = c_r = max = sum = _v; return; }
} T[_N << 3];

void _update(int p)
{
    T[p].c_l = max(T[p<<1].c_l, T[p<<1].sum+T[p<<1|1].c_l);
    T[p].c_r = max(T[p<<1|1].c_r, T[p<<1|1].sum+T[p<<1].c_r);
    T[p].max = max(max(T[p<<1].max, T[p<<1|1].max), T[p<<1].c_r+T[p<<1|1].c_l);
    T[p].sum = T[p<<1].sum + T[p<<1|1].sum;
    return;
}

void _init(int p, int x, int y)
{
    T[p].set_lr(x, y);
    if (x == y){ T[p].set_val(A[x]); return; }
    _init(p<<1, x, T[p].mid), _init(p<<1|1, T[p].mid+1, y);
    _update(p);
    return;
}

//返回 [x, y] 中从线段 p 左端开始的连续最大和 
int _begin_l(int p, int y)
{
    if (T[p].r <= y) return T[p].c_l;
    if (y <= T[p].mid) return _begin_l(p<<1, y);
    return max(_begin_l(p<<1, y), T[p<<1].sum+_begin_l(p<<1|1, y));
}

int _begin_r(int p, int x)
{
    if (T[p].l >= x) return T[p].c_r;
    if (x > T[p].mid) return _begin_r(p<<1|1, x);
    return max(_begin_r(p<<1|1, x), T[p<<1|1].sum+_begin_r(p<<1, x));
}

int _query(int p, int x, int y)
{
    if (x <= T[p].l && T[p].r <= y) return T[p].max;
    int maxx = -INF;
    if (x <= T[p].mid && y >= T[p].l) maxx = max(maxx, _query(p<<1, x, y));
    if (x <= T[p].r && y > T[p].mid) maxx = max(maxx, _query(p<<1|1, x, y));
    if (x <= T[p].mid && T[p].mid < y) maxx = max(maxx, _begin_r(p<<1, x)+_begin_l(p<<1|1, y));
    return maxx;
}

void _modify(int p, int x, int y)
{
    if (T[p].l == T[p].r) { T[p].set_val(y); return ; }
    if (x <= T[p].mid) _modify(p<<1, x, y);
    else _modify(p<<1|1, x, y);
    _update(p);
    return;
}

int main()
{
    int i;
    scanf("%d%d", &n, &m);
    for (i = 1; i <= n; ++i)
        scanf("%d", &A[i]);
    _init(1, 1, n);
    while (m--) {
        int ins, t1, t2;
        scanf("%d%d%d", &ins, &t1, &t2);
        if (ins == 1)
            printf("%d
", t1 <= t2 ? _query(1, t1, t2) : _query(1, t2, t1));
        else
            _modify(1, t1, t2);
    }
    return 0;
}