MnZn P3372线段树模板求条 玄关
  • 板块灌水区
  • 楼主LinkQR
  • 当前回复3
  • 已保存回复3
  • 发布时间2024/6/3 20:12
  • 上次更新2024/6/3 21:51:36
查看原帖
MnZn P3372线段树模板求条 玄关
1094385
LinkQR楼主2024/6/3 20:12
#include<bits/stdc++.h>
using namespace std;
long long a[100005],tree[400005],tag[400005];
void build(long long p,long long l,long long r){
	tag[p]=0;
	if(l==r){
		tree[p]=a[l];
		return;
	}
	long long mid=(l+r)/2;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	tree[p]=tree[p*2]+tree[p*2+1];
}
void addtag(long long p,long long pl,long long pr,long long k){
	tag[p]+=k;
	tree[p]+=k*(pr-pl+1);
}
void push_down(long long p,long long pl,long long pr){
	if(tag[p]){
		long long mid=(pl+pr)/2;
		addtag(p*2,pl,mid,tag[p]);
		addtag(p*2+1,mid+1,pr,tag[p]);
		tag[p]=0;
	}
}
void update(long long l,long long r,long long p,long long pl,long long pr,long long k){
	if(l<=pl&&pr<=r){
		addtag(p,pl,pr,k);
		return;
	}
	push_down(p,pl,pr);
	long long mid=(l+r)/2;
	if(l<=mid)update(l,r,p*2,pl,mid,k);
	if(r>mid)update(l,r,p*2+1,mid+1,pr,k);
	tree[p]=tree[p*2]+tree[p*2+1];
}
long long query(long long l,long long r,long long p,long long pl,long long pr){
	if(l<=pl&&pr<=r)return tree[p];
	push_down(p,pl,pr);
	long long res=0,mid=(pl+pr)/2;
	if(l<=mid)res+=query(l,r,p*2,pl,mid);
	if(mid<r)res+=query(l,r,p*2+1,mid+1,r);
	return res;
}
int main(){
	long long n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i];
	build(1,1,n);
	while(m--){
		long long op,l,r,k;
		cin>>op;
		if(op==1){
			cin>>l>>r>>k;
			update(l,r,1,1,n,k);
		}
		else{
			cin>>l>>r;
			cout<<query(l,r,1,1,n)<<endl;
		}
	}
	return 0;
}

过不去样例

2024/6/3 20:12
加载中...