树状数组模板题 & (树状数组 2:区间修改,单点查询)

题目链接

在这里插入图片描述
在这里插入图片描述

这是LibreOJ上的树状数组模板题,这题考查的是区间修改和单点查询,运用到了差分思想,在此附上一版AC代码:

#include<iostream>
using namespace std;
typedef long long ll;

ll b[1000005], bb[1000005], a[1000005];
//a是原数组,b是a的差分数组,bb是维护b的树状数组。
ll n, q;

ll lowbit(ll x){
	return x & (-x);
}

void creat(){
	for(ll i=1;i<=n;i++){
		bb[i]+=b[i];
		ll j=i+lowbit(i);
		if(j<=n) bb[j]+=bb[i];
	}
}
//建树。
ll getsum(ll x){
	ll ans=0;
	while(x>=1){
		ans+=bb[x];
		x-=lowbit(x);
	}
	return ans;
}
//求前缀和。
void add(ll x, ll k){
	while(x<=n){
		bb[x] += k;
		x += lowbit(x);
	}
}
//单点修改。区间修改因为差分思想可转变为俩个单点修改。
int main()
{
	scanf("%lld%lld", &n, &q);
	for(int i=1;i<=n;i++){
		scanf("%lld", &a[i]);
		if(i==1){
			b[1]=a[1];
		}else{
			b[i]=a[i]-a[i-1];
		}
	}
	creat();
	
	ll flag;
	ll l, r, x;
	ll y;
	for(int i=1;i<=q;i++){
		scanf("%lld", &flag);
		if(flag==2){
			scanf("%lld", &y);
			printf("%lld\n", getsum(y));
		}else {
			scanf("%lld%lld%lld", &l, &r, &x);
			add(l, x);
			if(r!=n){
				add(r+1, -x);
			}
		}
	}
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值