蓝桥 操作格子 线段树 单点更新 区间查询

算法训练 操作格子  

时间限制：1.0s   内存限制：256.0MB

       

问题描述

有n个格子，从左到右放成一排，编号为1-n。

共有m次操作，有3种操作类型：

1.修改一个格子的权值，

2.求连续一段格子权值和，

3.求连续一段格子的最大值。

对于每个2、3操作输出你所求出的结果。

输入格式

第一行2个整数n，m。

接下来一行n个整数表示n个格子的初始权值。

接下来m行，每行3个整数p,x,y，p表示操作类型，p=1时表示修改格子x的权值为y，p=2时表示求区间[x,y]内格子权值和，p=3时表示求区间[x,y]内格子最大的权值。

输出格式

有若干行，行数等于p=2或3的操作总数。

每行1个整数，对应了每个p=2或3操作的结果。

样例输入

4 3
1 2 3 4
2 1 3
1 4 3
3 1 4

样例输出

6
3

数据规模与约定

对于20%的数据n <= 100，m <= 200。

对于50%的数据n <= 5000，m <= 5000。

对于100%的数据1 <= n <= 100000，m <= 100000，0 <= 格子权值 <= 10000。

#include <bits/stdc++.h>

using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=100005;
typedef long long ll;

struct node{
	int l,r;
	ll Max,sum,lazy;
	void update(ll x){
		Max+=x;
		sum+=x;
		lazy+=x;
	}
}tree[maxn*4];

int n,m,a[maxn];

void push_up(int x){
	tree[x].sum=tree[x<<1].sum+tree[x<<1|1].sum;
	tree[x].Max=max(tree[x<<1].Max,tree[x<<1|1].Max);
}

void push_down(int x){
	int lazyval=tree[x].lazy;
	if(lazyval){
		tree[x<<1].update(lazyval);
		tree[x<<1|1].update(lazyval);
		tree[x].lazy=0;
	}
}

void build(int x,int l,int r){
	tree[x].l=l,tree[x].r=r;
	tree[x].sum=tree[x].Max=tree[x].lazy=0;
	if(l==r)
		tree[x].sum=tree[x].Max=0;
	else{
		int mid=(l+r)/2;
		build(x<<1,l,mid);
		build(x<<1|1,mid+1,r);
		push_up(x);
	}
}

void update(int x,int l,int r,int val){
	int L=tree[x].l,R=tree[x].r;
	//if(l==L&&r==R)	
	if(l<=L&&R<=r){
		tree[x].update(val);	
	}
	else{
		push_down(x);
		int mid=(L+R)/2;
		if(l<=mid)	update(x<<1,l,r,val);
		if(r>mid)	update(x<<1|1,l,r,val);
		push_up(x);
	}
}

ll querry_sum(int x,int l,int r){
	int L=tree[x].l,R=tree[x].r;
	if(l<=L&&R<=r){
		return tree[x].sum;	
	}
	else{
		push_down(x);
		int mid=(L+R)/2;
		ll ans=0;
		if(l<=mid)	ans+=querry_sum(x<<1,l,r);
		if(r>mid)	ans+=querry_sum(x<<1|1,l,r);
		push_up(x);
		return ans;
	}
}

ll querry_max(int x,int l,int r){
	int L=tree[x].l,R=tree[x].r;
	if(l<=L&&R<=r){
		return tree[x].Max;	
	}
	else{
		push_down(x);
		int mid=(L+R)/2;
		ll ans=-inf;
		if(l<=mid)	ans=max(ans,querry_max(x<<1,l,r));
		if(r>mid)	ans=max(ans,querry_max(x<<1|1,l,r));
		push_up(x);
		return ans;
	}
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	build(1,1,n);
	for(int i=1;i<=n;i++)
		update(1,i,i,a[i]);
		/*
	for(int i=1;i<=7;i++){
		printf("%d ",tree[i].Max);
	}
	printf("\n");
	*/
	for(int i=1;i<=m;i++){
		int q,u,v;
		scanf("%d%d%d",&q,&u,&v);
		if(q==1){
			update(1,u,u,v-a[u]);
			a[u]=v;
		}
		if(q==2){
			ll ans = querry_sum(1,u,v);
			printf("%lld\n",ans);
		}
		if(q==3){
			ll ans = querry_max(1,u,v);
			printf("%lld\n",ans);
		}
	}
	return 0;
}