洛谷 P4513 小白逛公园(线段树维护区间最大子段和)

传送门


解题思路

其实很板子。
但是我因为一开始写麻烦了又懒得修改所以出现了bug调了一节课。。。
线段树里存区间max,lmax,rmax,sum,意义跟变量名一样。
深刻体会到了pushup里采用取地址符调用的重要性。

AC代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=5e5+5;
int n,m;
struct node{
	int max,lmax,rmax,sum;
}d[maxn*5];
void pushup(int id,int lson,int rson){
	d[id].max=max(max(d[lson].max,d[rson].max),d[lson].rmax+d[rson].lmax);
	d[id].sum=d[lson].sum+d[rson].sum;
	d[id].lmax=max(d[lson].lmax,d[lson].sum+d[rson].lmax);
	d[id].rmax=max(d[rson].rmax,d[rson].sum+d[lson].rmax); 
}
void update(int id,int l,int r,int x,int v){
	if(l==r){
		d[id].max=d[id].lmax=d[id].rmax=d[id].sum=v;
		return;
	}
	int mid=(l+r)/2;
	if(x<=mid) update(id*2,l,mid,x,v);
	else update(id*2+1,mid+1,r,x,v);
	pushup(id,id*2,id*2+1);
}
node query(int id,int l,int r,int x,int y){
	if(x<=l&&r<=y){
		return d[id];
	}
	int mid=(l+r)/2;
	if(x<=mid){
		if(y<=mid) return query(id*2,l,mid,x,y);
		d[4*n+1]=query(id*2,l,mid,x,y);
		node aa=d[4*n+1]; 
		d[4*n+2]=query(id*2+1,mid+1,r,x,y);
		d[4*n+1]=aa;
		pushup(4*n,4*n+1,4*n+2);
		return d[4*n];
	}
	return query(id*2+1,mid+1,r,x,y);
}
int main(){
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int a;
		cin>>a;
		update(1,1,n,i,a);
	}
	for(int i=1;i<=m;i++){
		int k,p,s;
		cin>>k>>p>>s;
		if(k==1){
			if(p>s) swap(p,s); 
			cout<<query(1,1,n,p,s).max<<endl;
		}else{
			update(1,1,n,p,s);
		}
	}
	return 0;
}
posted @ 2021-09-16 17:23  尹昱钦  阅读(33)  评论(0编辑  收藏  举报