AcWing 245. 你能回答这些问题吗——线段树

题目链接

题目大意

给你一个长度为N的数列A，以及M条指令（N <= 5e5, M <= 1e5），每条指令可能是以下两种之一：

1. “1 x y”，如果x > y，则交互xy，查询区间[x, y]中最大连续子段和，即 m a x x   ⩽   l   ⩽   r   ⩽   y { ∑ i   =   l r A [ i ] } max{}_{{}x\ \leqslant \ l\ \leqslant \ r\ \leqslant \ y}\{ \sum^{r}_{i\ =\ l} A[i]\} maxx ⩽ l ⩽ r ⩽ y​{∑i = lr​A[i]}
2. “2 x y” 将下标为x的值改为y

解题思路

由于连续子段和具有线性的性质，且这题至少需要一个O(nlogn)的复杂度，我们考虑用线段树来维护区间最大连续子段和。
我们对于每个树上的节点要维护四个值
sum：区间和
lmaxn：紧靠区间左边的最大连续子段和
rmaxn：紧靠区间右边的最大连续子段和
dat：区间最大子段和
我们最关心的一件事就是如何pushup
对于区间和的更新，显然是左右子节点区间和
sum[rt] = sum[rt << 1] + sum[rt << 1 | 1]

对于lmaxn的更新，就是max(左子节点lmaxn, 左子节点区间和+右子节点lmaxn)
为什么是这样更新呢？当该节点的lmaxn不需要跨过左右子节点时，就是左子节点的lmaxn，当该节点的lmaxn需要跨过左右子节点时，该节点的lmaxn就是左子节点的所有点之和加上右子节点的lmaxn，大家可以好好理解一下这句话，这很妙。
lmaxn[rt] = max(lmaxn[rt << 1], sum[rt << 1] + lmaxn[rt << 1 | 1])

同理我们可以得出
rmaxn[rt] = max(rmaxn[rt << 1 | 1], sum[rt << 1 | 1] + rmaxn[rt << 1])

最后，如何更新dat呢？很容易想到，dat = max(左dat，右dat)，但是这是不对的 ，这样没有考虑跨越左右节点的连续子区间的情况，跨越左右两区间节点的最大连续子段和为 左rmaxn+右lmaxn，这样就是正确地转移了dat
dat[rt] = max(max(dat[rt >> 1], dat[rt >> 1 | 1]), rmaxn[rt >> 1] + lmaxn[rt >> 1 | 1]
ok这样我们就完成了pushup的操作
最后还有一个关于query的易错点，给出我的错误示例

int query(int L, int R, int l, int r, int rt){
	if(L <= l && r <= R)
		return tree[rt].dat;
	int m = (l + r) >> 1;
	int ret = -inf;
	if(L <= m)
		ret = max(ret, query(L, R, l, m, rt >> 1));
	if(m < R)
		ret = max(ret, query(L, R, m+1, r, rt >> 1 | 1));
	return ret;
}

一般来说query函数就是这样写，但是有一个问题，这样只考虑了左右子树的连续最大和，没有考虑跨越两个区间的最大连续子段和，所以我们要考虑连续区间的怎么办呢？我们知道树的节点包含了这一项信息，所以我们每次query的时候返回一个树的节点即可

代码

#include <bits/stdc++.h>
#define ll long long
#define qc ios::sync_with_stdio(false); cin.tie(0);cout.tie(0)
#define fi first
#define se second
#define PII pair<int, int>
#define PLL pair<ll, ll>
#define pb push_back
using namespace std;
const int MAXN = 5e5 + 7;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9 + 7;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
    while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
    return x*f;
}
struct node{
	int sum, lm, rm, dat;
	node(){
		sum = 0;
		lm = -inf;
		rm = -inf;
		dat = -inf;
	}
}tree[MAXN << 2];
void pushup(int rt){
	tree[rt].sum = tree[rt << 1].sum + tree[rt << 1 | 1].sum;
	tree[rt].lm = max(tree[rt << 1].lm, tree[rt << 1].sum + tree[rt << 1 | 1].lm);
	tree[rt].rm = max(tree[rt << 1 | 1].rm, tree[rt << 1 | 1].sum + tree[rt << 1].rm);
	tree[rt].dat = max(max(tree[rt << 1].dat, tree[rt << 1 | 1].dat), tree[rt << 1].rm + tree[rt << 1 | 1].lm);
}
void build(int l, int r, int rt){
	if(l == r){
		cin >> tree[rt].sum;
		tree[rt].lm = tree[rt].rm = tree[rt].dat = tree[rt].sum;
		return ;
	}
	int m = (l + r) >> 1;
	build(l, m, rt << 1);
	build(m+1, r, rt << 1 | 1);
	pushup(rt);
}
// int query(int L, int R, int l, int r, int rt){
// 	if(L <= l && r <= R)
// 		return tree[rt].dat;
// 	int m = (l + r) >> 1;
// 	int ret = -inf;
// 	if(L <= m)
// 		ret = max(ret, query(L, R, l, m, rt >> 1));
// 	if(m < R)
// 		ret = max(ret, query(L, R, m+1, r, rt >> 1 | 1));
// 	return ret;
// }
node query(int L, int R, int l, int r, int rt){
	if(L <= l && r <= R)
		return tree[rt];
	int m = (l + r) >> 1;
	node ret, lll, rr;
	if(L <= m)
		lll = query(L, R, l, m, rt << 1);
	if(m < R)
		rr = query(L, R, m+1, r, rt << 1 | 1);

	ret.sum = lll.sum + rr.sum;
	ret.lm = max(lll.lm, lll.sum + rr.lm);
	ret.rm = max(rr.rm, rr.sum + lll.rm);
	ret.dat = max(max(lll.dat, rr.dat), lll.rm + rr.lm);
	return ret;
}

void update(int L, int R, int x, int l, int r, int rt){
	if(L <= l && r <= R){
		tree[rt].lm = tree[rt].sum = tree[rt].rm = tree[rt].dat = x;
		return ;
	}
	int m = (l + r) >> 1;
	if(L <= m)
		update(L, R, x, l, m, rt << 1);
	if(m < R)
		update(L, R, x, m+1, r, rt << 1 | 1);
	pushup(rt);
}
int n, m;
int a[MAXN];
void solve(){
    cin >> n >> m;
    build(1, n, 1);
    while(m--){
    	int k, x, y;
    	cin >> k >> x >> y;
    	if(k == 1){
    		if(x > y)
    			swap(x, y);
    		cout << query(x, y, 1, n, 1).dat << endl;
    	}
    	else{
    		update(x, x, y, 1, n, 1);
    	}
    }
}

int main()
{
	#ifdef ONLINE_JUDGE
    #else
       freopen("in.txt", "r", stdin);
       freopen("out.txt", "w", stdout);
    #endif

	qc;
    int T;
    // cin >> T;
    T = 1;
    while(T--){

        solve();
    }
	return 0;
}