悬2观,求改
查看原帖
悬2观,求改
823785
chen0717楼主2024/7/24 11:30

代码如下:

#include<bits/stdc++.h>
#define ls rt<<1
#define rs rt<1|1
#define lson ls,l,mid
#define rson rs,mid+1,r
using namespace std;

long long n,m;
vector<long long> tr,a,lazy;

void push_up(long long rt){
	tr[rt]=tr[ls]+tr[rs];
}

void push_down(long long rt,long long l,long long r){
	if(lazy[rt]==0) return;
	long long mid=l+r>>1;
	lazy[ls]+=lazy[rt];
	lazy[rs]+=lazy[rt];
	tr[ls]+=lazy[rt]*(mid-l+1);
	tr[rs]+=lazy[rt]*(r-mid);
	lazy[rt]=0;
}

void build(long long rt,long long l,long long r){
	lazy[rt]=0;
	if(l==r){
		tr[rt]=a[l];
		return;
	}
	long long mid=l+r>>1;
	build(lson);
	build(rson);
	push_up(rt);
}
long long query(long long rt,long long l,long long r,long long ql,long long qr){
	if(ql<=l&&r<=qr){
		return tr[rt];
	}
	long long mid=l+r>>1;
	push_down(rt,l,r);
	if(qr<=mid) return query(lson,ql,qr);
	if(mid<ql) return query(rson,ql,qr);
	return query(lson,ql,qr)+query(rson,ql,qr);
}

void update(long long rt,long long l,long long r,long long ql,long long qr,long long k){
	if(ql<=l&&r<=qr){
		lazy[rt]+=k;
		tr[rt]+=k*(r-l+1);
		return;
	}
	long long mid=l+r>>1;
	push_down(rt,l,r);
	if(ql<=mid) update(lson,ql,qr,k);
	if(mid<qr) update(rson,ql,qr,k);
	push_up(rt);
}


int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n>>m;
	lazy=tr=vector<long long>(n*4+1,0);
	a=vector<long long>(n+1,0);
	for(long long i=1;i<=n;i++)
		cin>>a[i];
	build(1,1,n);
	for(long long i=1;i<=m;i++){
		long long op;
		cin>>op;
		if(op==1){
			long long l,r,x;
			cin>>l>>r>>x;
			update(1,1,n,l,r,x);
		}
		if(op==2){
			long long l,r;
			cin>>l>>r;
			cout<<query(1,1,n,l,r)<<'\n';
		}
	}
}
2024/7/24 11:30
加载中...