线段树1 爆零求助
查看原帖
线段树1 爆零求助
442963
wu13115899522楼主2022/8/22 09:32
#include<bits/stdc++.h>
using namespace std;
long long a[100005],t[400005],v[400004];
int n,m;
void build(int k,int l,int r){
	if(l==r){
		t[k]=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	t[k]=t[k<<1]+t[k<<1|1];
}
void change(int k,int l,int r,int x,int y,long long z){
	if(l==x&&r==y){
		v[k]+=z;
		return;
	}
	if(v[k]){
		v[k<<1]+=v[k],v[k<<1|1]+=v[k],v[k]=0;
	}
	int mid=(l+r)>>1;
	if(y<=mid)change(k<<1,l,mid,x,y,z);
	else if(x>mid)change(k<<1|1,mid+1,r,x,y,z);
	else change(k<<1,l,mid,x,mid,z),change(k<<1|1,mid+1,r,mid+1,y,z);
	t[k]+=t[k<<1]+v[k<<1]*(mid-l+1)+t[k<<1|1]+v[k<<1|1]*(r-mid);
} 
int get(int k,int l,int r,int x,int y){
	if(l==x&&r==y){
		return t[k];
	} 
	if(v[k]){
		v[k<<1]+=v[k],
		v[k<<1|1]+=v[k],v[k]=0;
	}
	int mid=(l+r)>>1;
	long long res=0;
	if(y<=mid)res=get(k<<1,l,mid,x,y);
	if(x>mid)res=get(k<<1|1,mid+1,r,x,y);
	if(x<=mid&&y>mid)res= et(k<<1,l,mid,x,mid)+get(k<<1|1,mid+1,r,mid+1,y);
	t[k]+=t[k<<1]+v[k<<1]*(mid-l+1)+t[k<<1|1]+v[k<<1|1]*(r-mid);
	return res;
}
signed 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<=m;i++){
		int s;
		scanf("%d",&s);
		if(s==1){
			int x,y;long long z;
			scanf("%d%d%lld",&x,&y,&z);
			change(1,1,n,x,y,z);
		}
		else {
			int x,y;
			scanf("%d%d",&x,&y);
			cout<<get(1,1,n,x,y)<<endl;
		}
	}
	return 0;
}
2022/8/22 09:32
加载中...