原题链接
给定长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:
1 x y,查询区间 [x,y] 中的最大连续子段和,即 maxx≤l≤r≤y{∑i=lrA[i]}。
2 x y,把 A[x] 改成 y。
对于每个查询指令,输出一个整数表示答案。
输入格式
第一行两个整数 N,M。
第二行 N 个整数 A[i]。
接下来 M 行每行 3 个整数 k,x,y,k=1 表示查询(此时如果 x>y,请交换 x,y),k=2 表示修改。
输出格式
对于每个查询指令输出一个整数表示答案。
每个答案占一行。
数据范围
N≤500000,M≤100000,
−1000≤A[i]≤1000
输入样例:
5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 3 2
输出样例:
2
-1
线段树敲一次错一次,这次终于一次性成功了
#include <iostream>
#include <cstring>
using namespace std;
const int N = 500010;
int n, m;
int w[N];
struct node {
int l, r;
int sum, lmax, rmax, tmax;
} tr[4 * N];
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 = l, tr[u].r = r;
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
pushup(u);
}
}
void modify(int u, int x, int v) {
if (tr[u].l == x && tr[u].r == x) {
tr[u] = {x, x, v, v, v, v};
return;
} 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];
else {
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() {
cin >> n >> m;
for (int i = 1; i <= n; i++)
scanf("%d", &w[i]);
build(1, 1, n);
int k, x, y;
while (m--) {
cin >> k >> x >> y;
if (k == 1) {
if (x > y)
swap(x, y);
cout << query(1, x, y).tmax << endl;
}
else
modify(1, x, y);
}
}