社区讨论

20pts求调

P3368【模板】树状数组 2参与者 3已保存回复 7

讨论操作

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

当前回复
7 条
当前快照
1 份
快照标识符
@mhjusa6t
此快照首次捕获于
2025/11/04 08:51
4 个月前
此快照最后确认于
2025/11/04 08:51
4 个月前
查看原帖
请看记录:https://www.luogu.com.cn/record/212212311
代码:
CPP
#include<iostream>
#define int long long 
using namespace std; 
const int N=5e5+10;
int a[N],ans=0;
struct node{
	int l,r,sum;
}tree[N*4];
void build(int i,int l,int r){
	tree[i].l=l;
	tree[i].r=r;
	if(l==r){
		tree[i].sum=a[l];
		return ;
	}
	int mid=(l+r)/2;
	build(i*2,l,mid);
	build(i*2+1,mid+1,r);
}
int search(int i,int l,int r){
	if(tree[i].l>=l && tree[i].r<=r) return tree[i].sum;
	if(tree[i].r<l || tree[i].l>r) return 0;
	int s=0;
	if(tree[i*2].r>=l) s+=search(i*2,l,r);
	if(tree[i*2+1].l<=r) s+=search(i*2+1,l,r);
	return s;
}
void add(int i,int idx,int k){
	if(tree[i].l==tree[i].r){ 
		tree[i].sum+=k;
		return ;
	}
	if(idx<=tree[i*2].r) add(i*2,idx,k);
	else add(i*2+1,idx,k);
	tree[i].sum=tree[i*2].sum+tree[i*2+1].sum; 
}
void modify(int i,int l,int r,int k){
	if(tree[i].l>=l && tree[r].r<=r){
		tree[i].sum+=k;
		return ;
	}
	int mid=(tree[i].l+tree[i].r)/2;
	if(l<=mid) modify(i*2,l,r,k);
	if(r>mid) modify(i*2+1,l,r,k);
}
void query(int p,int x){
	ans+=tree[p].sum;
	if(tree[p].l==tree[p].r) return ;
	int mid=(tree[p].l+tree[p].r)/2;
	if(x<=mid) query(p*2,x);
	else query(p*2+1,x);
}
signed main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,1,n);
	for(int i=1;i<=m;i++){
		int ops,x,y,k;
		cin>>ops>>x;
		if(ops==1){
			cin>>y>>k;
			modify(1,x,y,k); 
		}
		else{
			ans=0;
			query(1,x);
			cout<<ans<<endl;
		}
	}
	return 0;
}

回复

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

正在加载回复...