给定长度为 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<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'
using namespace std;
typedef pair<int, int> PII;
typedef long long ll;
const int N = 500010;
int n, m;
int a[N];
struct Node
{
int l, r;
int sum, lmax, rmax, tmax;
}tr[N * 4];
inline 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);
}
inline void pushup(int u)
{
pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}
inline void build(int u, int l, int r)
{
if(l == r)tr[u] = {l, r, a[l], a[l], a[l], a[l]};
else
{
tr[u] = {l, r};//不要忘了这一步
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
pushup(u);
}
}
inline void modify(int u, int x, int c)
{
if(tr[u].l == x && tr[u].r == x)tr[u] = {x, x, c, c, c, c};//记得每次修改要把所有值都修改到
else
{
int mid = tr[u].l + tr[u].r >> 1;
if(x <= mid)modify(u << 1, x, c);
else modify(u << 1 | 1, x, c);
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(l > mid)return query(u << 1 | 1, l, r);
else if(r <= mid)return query(u << 1, l, r);
else
{
Node left = query(u << 1, l, r);
Node right = query(u << 1 | 1, l, r);
Node res;
pushup(res, left, right);//这里的pushup并不是合并树中节点,只是借用一下这个函数计算答案
return res;
}
}
}
int main()
{
IOS
int n, m;
cin >> n >> m;
for(int i = 1; i <= n; i ++)cin >> a[i];
build(1, 1, n);
while(m --)
{
int op, x, y;
cin >> op >> x >> y;
if(op == 1)
{
if(x > y)swap(x, y);
cout << query(1, x, y).tmax << endl;
}
else
{
modify(1, x, y);
}
}
return 0;
}
一个区间的最大连续子段和考虑三个方面:只在左孩子区间那一部分的最大连续子段和、只在右孩子区间那一部分的最大连续子段和、左右孩子区间都涉及的最大连续子段和。
第三种情况可以用左区间的最大后缀和+右区间的最大前缀和来实现;
对于一个区间最大前缀和,可以在当前区间左孩子区间的最大前缀和 and 左孩子区间和+右孩子最大前缀和中取一个最大值来实现。
最大后缀和同理。
另外发现一个小细节:
u.tmax = max(max(l.tmax, r.tmax), l.rmax + r.lmax);
u.tmax = max({l.tmax, r.tmax, l.rmax + r.lmax});
在pushup操作中像前者这样写比像后者这样写快了400ms。