小白逛公園(線段樹)

題目描述

小新經常陪小白去公園玩，也就是所謂的遛狗啦…在小新家附近有一條“公園路”，路的一邊從南到北依次排着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;
}