线段树求调,实在看不出来了(玄关
查看原帖
线段树求调,实在看不出来了(玄关
1328469
glass_goldfish楼主2024/12/12 20:44
#include<bits/stdc++.h>
#define int long long
#define INF 0x3f3f3f
using namespace std;
int n,q,a[300001];
int w[300001],lzy[300001];
void build(int u,int l,int r){
	if(l==r)w[u]=a[u];
	else{
		int mid=(l+r)>>1;
		build(2*u,l,mid);build(2*u+1,mid+1,r);
		w[u]=w[2*u]+w[2*u+1];
	}
}
void update(int u,int nl,int nr,int l,int r,int k){
	if(nl>r||nr<l)return;
	else if(nl>=l&&nr<=r){
		lzy[u]+=k;
		w[u]+=(nl-nr+1)*k;
	}
	else{
		int mid=(nl+nr)>>1;
		lzy[u*2]+=lzy[u];
		w[u*2]+=(lzy[u]*(mid-nl+1));
		lzy[u*2+1]+=lzy[u];
		w[u*2+1]+=(lzy[u]*(nr-mid+1));
		lzy[u]=0;
		update(u*2,nl,mid,l,r,k);
		update(u*2+1,mid+1,nr,l,r,k);
		w[u]=w[u*2]+w[u*2+1];
	}
}
int query(int u,int nl,int nr,int l,int r){
	if(nl>r||nr<l)return 0;
	else if(nl>=l&&nr<=r)return w[u];
	else{
		int mid=(nl+nr)>>1;
		lzy[u*2]+=lzy[u];
		w[u*2]+=(lzy[u]*(mid-nl+1));
		lzy[u*2+1]+=lzy[u];
		w[u*2+1]+=(lzy[u]*(nr-mid+1));
		lzy[u]=0;
		int p=query(u*2,nl,mid,l,r)+query(u*2+1,mid+1,nr,l,r);
		return p;
	}
}
signed main(){
	cin>>n>>q;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	build(1,1,n);
	while(q--){
		int opt,x,y,k;
		cin>>opt>>x>>y;
		if(opt==1){
			cin>>k;
			update(1,1,n,x,y,k);
		}
		else{
			cout<<query(1,1,n,x,y)<<"\n";
		} 
	}
	return 0;
} 
2024/12/12 20:44
加载中...