mxqz线段树1模板
查看原帖
mxqz线段树1模板
398132
爱好MC的蒟蒻楼主2021/8/26 21:49

rt

代码

#include <iostream>
#define int long long
using namespace std;
int n,m,x,y,z,op,d[100005],a[100005],tag[100005];
void build(int x,int y,int z){
	if(x==y){
		d[z]=a[x];
		return ;
	}
	int q=(x+y)/2;
	build(x,q,z*2);
	build(q+1,y,z*2+1);
	d[z]=d[z*2]+d[z*2+1];
}
void update(int l,int r,int al,int ar,int z,int id){
	if(l<=al&&r>=ar){
		d[id]+=(ar-al+1)*z;
		tag[id]+=z;
		return ;
	}
	int q=(al+ar)/2;
	if(tag[id]!=0){
		d[id*2]+=tag[id]*(q-al+1);
		d[id*2+1]+=tag[id]*(ar-q);
		tag[id*2]+=tag[id];
		tag[id*2+1]+=tag[id];
		tag[id]=0;
	}
	if(l<=q) update(l,r,al,q,z,id*2);
	if(r>q) update(l,r,q+1,ar,z,id*2+1);
	d[id]=d[id*2]+d[id*2+1];
}
int query(int l,int r,int al,int ar,int id){
	if(l<=al&&r>=ar){
		return d[id];
	}
	int q=(al+ar)/2,sum=0;
	if(tag[id]!=0){
		d[id*2]+=d[id]*(q-l+1);
		d[id*2+1]+=d[id]*(r-q);
		tag[id*2]+=tag[id];
		tag[id*2+1]+=tag[id];
		tag[id]=0;
	}
	if(l<=q) sum+=query(l,r,al,q,id*2);
	if(r>q) sum+=query(l,r,q+1,ar,id*2+1);
	return sum;
}
signed main(){
	cin >>n>>m;
	for(int i=1;i<=n;i++){
		cin >>a[i];
	}
	build(1,n,1);
	for(int i=1;i<=m;i++){
		cin >>op;
		if(op==1){
			cin >>x>>y>>z;
			update(x,y,1,n,z,1);
		}
		else{
			cin >>x>>y;
			cout <<query(x,y,1,n,1)<<endl;
		}
	}
	return 0;
}

样例最后一行不对,求巨佬查错

你可以获得关注+2

2021/8/26 21:49
加载中...