P4513 小白逛公园（线段树）

Posted 2021-08-08 SSL_LKJ

小白逛公园

题目传送门

输入样例

5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 2 3

输出样例

2
-1

解题思路

用线段树的每个节点维护：sum（区间和），maxx（区间最大子段和），lmax（从区间左端点开始的最大子段和），rmax（从区间右端点开始的最大子段和）

AC

#include<iostream> 
#include<cstdio>
using namespace std;
int n,m,a[500005];
struct node 
{
	int sum,l,r,maxx;
}f[5000005];
node add(node x,node y)//转移公式
{
	node z;
	z.sum=x.sum+y.sum;
	z.l=max(x.l,x.sum+y.l);
	z.r=max(y.r,y.sum+x.r);
	z.maxx=max(max(x.maxx,y.maxx),x.r+y.l);
	return z;
}
void pre(int k,int l,int r)
{
	if(l==r)//赋值
	{
		f[k].sum=f[k].l=f[k].r=f[k].maxx=a[l];
		return;
	}
	int mid=(l+r)/2;
	pre(k*2,l,mid);
	pre(k*2+1,mid+1,r);
	f[k]=add(f[k*2],f[k*2+1]);合并
}
node query(int k,int x,int y,int l,int r)
{
	if(x<=l&&r<=y)return f[k];
	int mid=(l+r)/2;
	if(y<=mid)return query(k*2,x,y,l,mid);
	if(mid+1<=x)return query(k*2+1,x,y,mid+1,r);
	node A,B,C;
	A=query(k*2,x,y,l,mid);
	B=query(k*2+1,x,y,mid+1,r);
	C=add(A,B);
	return C;	
} 
void change(int k,int x,int l,int r,int d)
{
	if(l==r)
	{
		f[k]=(node){d,d,d,d};
		return;
	}
	int mid=(l+r)/2;
	if(x<=mid)change(k*2,x,l,mid,d);
	if(mid+1<=x)change(k*2+1,x,mid+1,r,d);
	f[k]=add(f[k*2],f[k*2+1]);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	pre(1,1,n);
	while(m--)
	{
		int K,A,B;
		scanf("%d%d%d",&K,&A,&B);
		if(K==1)
		{
			if(A>B)swap(A,B);
			node ans=query(1,A,B,1,n);
			printf("%d\\n",ans.maxx);
		}
		else change(1,A,1,n,B);
	}
	return 0;
}

谢谢