CHOJ 4301【线段树+区间最大子段和】
描述
给定长度为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
数据范围与约定
- 对于100%的数据: N≤500000, M≤100000, |Ai|<=1000
题解：
#include
#include
#include
#include
#include
#define INF -0x3f3f3f3f
#define ll long long
using namespace std;
const int maxn = 500010;
int n, m;
int a[maxn];
struct SegmentTree{
int l, r;
int dat, sum, lmax, rmax;
}t[maxn<<2];
void init(SegmentTree &tree, int x){
tree.dat=tree.lmax=tree.rmax=tree.sum=x;
}
void push_up(int p){
t[p].sum = t[2*p].sum + t[2*p+1].sum;
t[p].lmax = max(t[2*p].lmax, t[2*p].sum+t[2*p+1].lmax);
t[p].rmax = max(t[2*p+1].rmax, t[2*p+1].sum+t[2*p].rmax);
t[p].dat = max(t[2*p].dat, max(t[2*p+1].dat, t[2*p].rmax+t[2*p+1].lmax));
}
void build(int p, int l, int r){
t[p].l = l, t[p].r = r;
if(l == r){
init(t[p], a[l]);
return;
}
int mid = (l+r)/2;
build(2*p, l, mid);
build(2*p+1, mid+1, r);
push_up(p);
}
void change(int p, int x, int val){
if(t[p].l == t[p].r){
init(t[p], val);
return ;
}
int mid = (t[p].l+t[p].r)/2;
if(x <= mid) change(2*p, x, val);
else change(2*p+1, x, val);
push_up(p);
}
SegmentTree ask(int p, int l, int r){
if(l <= t[p].l && r >= t[p].r){
return t[p];
}
int mid = (t[p].l+t[p].r)/2;
SegmentTree a, b, c;
init(a, INF), init(b, INF);
c.sum = 0;
if(l <= mid) {
a = ask(2*p, l, r);
c.sum += a.sum;
}
if(r > mid) {
b = ask(2*p+1, l, r);
c.sum += b.sum;
}
c.dat = max(max(a.dat, b.dat), a.rmax+b.lmax);
c.lmax = max(a.lmax, b.lmax+a.sum);
c.rmax = max(b.rmax, a.rmax+b.sum);
return c;
}
int main()
{
int k, x, y;
cin >> n >> m;
for(int i = 1; i <= n; i++){
cin >> a[i];
}
build(1,1,n);
while(m--){
cin >> k >> x >> y;
if(k == 1){
if(x > y) swap(x, y);
printf("%d\n", ask(1,x,y).dat);
}else change(1,x,y);
}
return 0;
}