小白逛公园(线段树)

题目描述

小新经常陪小白去公园玩，也就是所谓的遛狗啦…在小新家附近有一条“公园路”，路的一边从南到北依次排着n个公园，小白早就看花了眼，自己也不清楚该去哪些公园玩了。
    一开始，小白就根据公园的风景给每个公园打了分-.-。小新为了省事，每次遛狗的时候都会事先规定一个范围，小白只可以选择第a个和第b个公园之间（包括a、b两个公园）选择 连续 的一些公园玩。小白当然希望选出的公园的分数总和尽量高咯。同时，由于一些公园的景观会有所改变，所以，小白的打分也可能会有一些变化。
    那么，就请你来帮小白选择公园吧。

输入

第一行，两个整数N和M，分别表示表示公园的数量和操作（遛狗或者改变打分）总数。
    接下来N行，每行一个整数，依次给出小白 开始时对公园的打分。
    接下来M行，每行三个整数。第一个整数K，1或2。K=1表示，小新要带小白出去玩，接下来的两个整数a和b给出了选择公园的范围（1≤a,b≤N(a可能大于b)）；K=2表示，小白改变了对某个公园的打分，接下来的两个整数p和s，表示小白对第p个公园的打分变成了s（1≤p≤N）。
    其中，1≤N≤500 000，1≤M≤100 000，所有打分都是绝对值不超过1000的整数。

输出

小白每出去玩一次，都对应输出一行，只包含一个整数，表示小白可以选出的公园得分和的最大值。

样例输入

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

样例输出

2

-1

题意同山海经，只是多了点修改，不用再维护答案的左右下标，变得更好做了。

#include<cstdio>
#include<iostream>
#include<algorithm>
#define maxn 2000005
using namespace std;
int n,m,inf=0xfffffff;
struct tree{int lc,rc,sum,lmax,rmax,ma;tree(){lmax=rmax=ma=-inf;sum=0;}}tr[maxn];
void pushup(int x)
{
	int lch=x<<1,rch=lch|1;
	tr[x].sum=tr[lch].sum+tr[rch].sum;
	if(tr[lch].lmax<tr[lch].sum+tr[rch].lmax) tr[x].lmax=tr[lch].sum+tr[rch].lmax;
	else tr[x].lmax=tr[lch].lmax;
	if(tr[rch].rmax<tr[rch].sum+tr[lch].rmax) tr[x].rmax=tr[rch].sum+tr[lch].rmax;
	else tr[x].rmax=tr[rch].rmax;
	if(tr[lch].ma<tr[rch].ma)
	{
		if(tr[rch].ma<=tr[lch].rmax+tr[rch].lmax) tr[x].ma=tr[lch].rmax+tr[rch].lmax;
		else tr[x].ma=tr[rch].ma;
	}
	else if(tr[lch].ma<tr[lch].rmax+tr[rch].lmax) tr[x].ma=tr[lch].rmax+tr[rch].lmax;
	else tr[x].ma=tr[lch].ma;
}
void cmp(tree x,tree y,tree &z)
{
	z.sum=x.sum+y.sum;
	if(x.lmax<x.sum+y.lmax) z.lmax=x.sum+y.lmax;
	else z.lmax=x.lmax;
	if(y.rmax<y.sum+x.rmax) z.rmax=y.sum+x.rmax;
	else z.rmax=y.rmax;
	if(x.ma<y.ma)
	{
		if(y.ma<=x.rmax+y.lmax) z.ma=x.rmax+y.lmax;
		else z.ma=y.ma;
	}
	else if(x.ma<x.rmax+y.lmax) z.ma=x.rmax+y.lmax;
	else z.ma=x.ma;
}
void build(int x,int y,int z)
{
	tr[z].lc=x;tr[z].rc=y;
	if(x==y)
	{
		int tmp;scanf("%d",&tmp);
		tr[z].sum=tr[z].lmax=tr[z].rmax=tr[z].ma=tmp;
		return;
	}
	int mid=x+y>>1,lch=z<<1,rch=lch|1;
	build(x,mid,lch);build(mid+1,y,rch);
	pushup(z);
}
void updata(int x,int y,int z)
{
	if(tr[z].lc==x&&tr[z].rc==x){tr[z].sum=tr[z].lmax=tr[z].rmax=tr[z].ma=y;return;}
	int mid=tr[z].lc+tr[z].rc>>1,lch=z<<1,rch=lch|1;
	if(x<=mid) updata(x,y,lch);
	else updata(x,y,rch);
	pushup(z);
} 
tree query(int x,int y,int z)
{
	if(tr[z].lc>=x&&tr[z].rc<=y)
		return tr[z];
	int mid=tr[z].lc+tr[z].rc>>1,lch=z<<1,rch=lch|1;tree t1,t2,t3;
	if(x<=mid) t1=query(x,y,lch);
	if(mid<y) t2=query(x,y,rch);
	t1.lc=x;t1.rc=mid;t2.lc=mid+1;t2.rc=y;
	cmp(t1,t2,t3);
	return t3;
}
int main()
{
	scanf("%d%d",&n,&m);
	build(1,n,1);int x,y,z;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		if(x==1){if(y>z) swap(y,z); tree tmp=query(y,z,1);printf("%d\n",max(tmp.lmax,max(tmp.rmax,tmp.ma)));}
		else updata(y,z,1); 
	}
	return 0;
}