AcWing.245. 你能回答这些问题吗

AcWing.245. 你能回答这些问题吗

题目:

给定长度为N的数列A,以及M条指令,每条指令可能是以下两种之一:

1、“1 x y”,查询区间 [x,y] 中的最大连续子段和,即 maxx≤l≤r≤y{∑ri=l A[i]}。

2、“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


思路:线段树
节点信息:

l,r,区间和sum,区间最大连续子段和ans,包括左端点的最大连续子段和lx,包括右端点的最大连续子段和rx

用子节点更新父节点:
t[p].sum = t[p*2].sum + t[p*2+1].sum;
t[p].lx = max(t[p*2].lx, t[p*2].sum + t[p*2+1].lx);
t[p].rx = max(t[p*2+1].rx, t[p*2+1].sum + t[p*2].rx);
t[p].ans = max(max(t[p*2].ans, t[p*2+1].ans), t[p*2].rx + t[p*2+1].lx);

中文翻译:

sum:略
lx:
Alt
rx:同上

sum:
1.左子树的最大连续子段和 或 右子树的最大连续子段和
2.Alt

求解

给定一段区间[l, r],需要化为两个区间的和
由sum的求解过程可知,我们需要这两个区间的sum和lx,rx
但这两个区间不一定为某个节点,因此我们需要建立新的节点,节点信息由以上过程递归求出

假如现要求节点q,范围[l, r],现在来到树上节点p
设节点l为p的左子树上属于[l, r]的部分,节点r为p的右子树上属于[l, r]的部分
那么q的信息如上求

要注意的是,如果p的左子树与[l, r]无交集,即l为空,直接将r赋给q
r为空同理


代码
#include <iostream>
#include <cstdio>
using namespace std;

#define N 500001 
#define INF 1<<30
int n, m;
int a[N];
struct tree
{
	int l, r;
	int sum, lx, rx, ans;
	#define l(x) x.l
	#define r(x) x.r
	#define sum(x) x.sum
	#define lx(x) x.lx
	#define rx(x) x.rx
	#define ans(x) x.ans
}t[4*N];

inline void build(int p, int l, int r)
{
	l(t[p]) = l, r(t[p]) = r;
	if(l == r)
	{
		sum(t[p]) = lx(t[p]) = rx(t[p]) = ans(t[p]) = a[l];
		return ;
	}
	int mid = (l+r) / 2;
	build(2*p, l, mid);
	build(2*p+1, mid+1, r);
	sum(t[p]) = sum(t[2*p]) + sum(t[2*p+1]);
	lx(t[p]) = max(lx(t[2*p]), sum(t[2*p]) + lx(t[2*p+1]));
	rx(t[p]) = max(rx(t[2*p+1]), sum(t[2*p+1]) + rx(t[2*p]));
	ans(t[p]) = max( max( ans(t[2*p]), ans(t[2*p+1]) ), rx(t[2*p]) + lx(t[2*p+1]) );
}
inline void change(int p, int x, int v)
{
	int l = l(t[p]), r = r(t[p]);
	if(l == r)
	{
		sum(t[p]) = lx(t[p]) = rx(t[p]) = ans(t[p]) = v;
		return ;
	}
	int mid = (l+r) / 2;
	if(x <= mid)
		change(2*p, x, v);
	if(x > mid)
		change(2*p+1, x, v);
	sum(t[p]) = sum(t[2*p]) + sum(t[2*p+1]);
	lx(t[p]) = max(lx(t[2*p]), sum(t[2*p]) + lx(t[2*p+1]));
	rx(t[p]) = max(rx(t[2*p+1]), sum(t[2*p+1]) + rx(t[2*p]));
	ans(t[p]) = max( max( ans(t[2*p]), ans(t[2*p+1]) ), rx(t[2*p]) + lx(t[2*p+1]) );
}
tree ask(int p, int ll, int rr)
{
	if(ll <= l(t[p]) && rr >= r(t[p]))
		return t[p];
	int mid = (l(t[p]) + r(t[p])) / 2;
	tree l, r, q;
	sum(l) = lx(l) = rx(l) = ans(l) = sum(r) = lx(r) = rx(r) = ans(r) = -INF;
	sum(q) = 0;//q的值不能乱赋啊 QAQ 
	if(ll <= mid )
	{
		l = ask(2*p, ll, rr);
		sum(q) += sum(l);
	}
	if(rr > mid)
	{
		r = ask(2*p+1, ll, rr);
		sum(q) += sum(r);
	}
	ans(q) = max( max(ans(l), ans(r)), rx(l) + lx(r));//ans(p)没赋初值 不能乱用 
	lx(q) = max(lx(l), sum(l) + lx(r));
	rx(q) = max(rx(r), sum(r) + rx(l));
	if(ll > mid)
		lx(q) = max(lx(q), lx(r));
	//如果[ll, rr]与左子树无交集 那么q就在p的右子树中 即c=r 
	//ans, sum, rx 都已经赋过r的值(此时l的所有值都负无穷) 
	if(rr <= mid)
		rx(q) = max(rx(q), rx(l));
	return q; 
}

int main(){
	scanf("%d %d", &n, &m);
	for(int i = 1; i <= n; ++i)
	{
		scanf("%d", &a[i]);
	}
	build(1, 1, n);
	int k, x, y;
	while(m--)
	{
		scanf("%d %d %d", &k, &x, &y);
		if(k == 1)
		{
			if(x > y)
				swap(x, y);
			printf("%d\n", ans(ask(1, x, y)));
		}
		else
		{
			change(1, x, y);
			/*for(int i = 1; i <= n; ++i)
			{
				printf("%d %d %d %d %d\n", i, l(t[i]), r(t[i]), ans(t[i]), sum(t[i]));
			}	
			printf("\n");*/
		}
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值