求助线段树0分
查看原帖
求助线段树0分
556362
Unnamed114514楼主2021/10/5 08:00
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e6;
int n,m,a[maxn],add[maxn*4],sum[maxn*4];
void Build(int k,int l,int r){
	if(l==r){
		sum[k]=a[l];
		return;
	}
	int mid=l+r>>1;
	Build(k<<1,l,mid);
	Build(k<<1|1,mid+1,r);
	sum[k]=sum[k<<1]+sum[k<<1|1];
}
void Add(int k,int l,int r,int v){
	add[k]+=v;
	sum[k]+=v*(r-l+1);
}
void pushdown(int k,int l,int r,int mid){
	if(add[k]==0)
		return;
	Add(k<<1,l,mid,add[k]);
	Add(k<<1|1,mid+1,r,add[k]);
	add[k]=0;
}
int query(int k,int l,int r,int x,int y){
	if(l>=x&&r<=y)
		return sum[k];
	int mid=(l+r)>>1;
	int res=0;
	pushdown(k,l,r,mid);
	if(x<=mid)
		res+=query(k<<1,l,mid,x,y);
	if(mid<y)
		res+=query(k<<1|1,mid+1,r,x,y);
	return res;
}
void modify(int k,int l,int r,int x,int y,int v){
	if(l>=x&&r<=y)
		return Add(k,l,r,v);
	int mid=(l+r)>>1;
	pushdown(k,l,r,mid);
	if(x<=mid)
		modify(k<<1,l,mid,x,y,v);
	if(mid<y)
		modify(k<<1|1,mid+1,r,x,y,v);
	sum[k]=sum[k<<1]+sum[k<<1|1];
}
signed main(){
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	Build(1,1,n);
	while(m--){
		int a,b,c,d;
		cin>>a>>b>>c;
		if(a==1){
			cin>>d;
			modify(1,1,n,a,b,d);
		} else
			printf("%lld\n",query(1,1,n,a,b));
		
	}
	return 0;
}
2021/10/5 08:00
加载中...