社区讨论

10分求调

P3372【模板】线段树 1参与者 5已保存回复 11

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
11 条
当前快照
1 份
快照标识符
@lymzgbo5
此快照首次捕获于
2024/07/15 20:50
2 年前
此快照最后确认于
2024/07/15 21:57
2 年前
查看原帖
样例过了,但只有10分
CPP
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=4e5+10;
int n,m;
int w[N],lz[N],a[N];
void build(int id,int l,int r ){
	if(l==r){
	w[id]=a[l];
	return;	
	}
	int mid=l+r>>1;
	build(id<<1,l,mid);
	build(id<<1|1,mid+1,r);
	w[id]=w[id<<1]+w[id<<1|1];
}



void push_down(int id,int l,int r,int mid){
	if(lz[id]!=0){
		w[id<<1]+=(mid-l+1)*lz[id];
		w[id<<1|1]+=(r-mid+1+1)*lz[id];
		lz[id<<1]+=lz[id];
		lz[id<<1|1]+=lz[id];
		lz[id]=0;
	}
}
void update(int id,int l,int r,int L,int R,int k){
	if(L<=l&&R>=r){
		w[id]+=(r-l+1)*k;
		lz[id]+=k;
		return;
	}
	int mid=l+r>>1;
	push_down(id,l,r,mid);
	if(L<=mid) update(id<<1,l,mid,L,R,k);
	if(R>mid) update(id<<1|1,mid+1,r,L,R,k);
	w[id]=w[id<<1]+w[id<<1|1];
}

int find(int id,int l,int r,int L,int R){
	int res=0;
	if(L<=l&&r<=R) return w[id];
	int mid=l+r>>1;
	push_down(id,l,r,mid);
	if(L<=mid) res+=find(id<<1,l,mid,L,R);
	if(R>mid) res+=find(id<<1|1,mid+1,r,L,R);
//	w[id]=w[id<<1]+w[id<<1|1];
	return res;
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	build(1,1,n);
	while(m--){
		int x,y,op,k;
		cin>>op;
		if(op==2){
			cin>>x>>y;
		cout<<find(1,1,n,x,y)<<endl;
		}
		else{
			cin>>x>>y>>k;
			update(1,1,n,x,y,k);
		}
	}
}

回复

11 条回复,欢迎继续交流。

正在加载回复...