树状数组 - 模板

这篇博客探讨了如何使用位运算技巧优化数组的区间修改和查询操作。通过低字节位运算(lowbit)实现单点修改和区间查询,以及在差分数组上的区间修改和单点查询。文章提供了基础代码示例,展示了如何更新和获取数组的前缀和,以提高算法效率。
摘要由CSDN通过智能技术生成
深入学习链接:点击进入
数组及函数

c [ i ] = a ( i - 2 ^ k + 1 ) + … + a [ i ]( 这里 c 数组是根据 a 数组更新的,实现不同的功能,a 数组的含义不同 )
( 设节点编号为 i ,那么这个节点管辖的区间有 2 ^ k( 其中 k 为 i 二进制末尾 0 的个数)个元素,且最后一个元素为 a [ i ] 。)
lowbit ( i ) => i & -i 计算 i 对应的 2 ^ k
add 函数更新 i 这位以及它之后的包含它的区间对应的 c 数组
getsum 函数求 1 - i 的和

单点修改 + 区间查询

对于单点修改,c [ i ] 依据 a [ i ] 原数组进行更新。

基本代码
int n,m,a[maxn],c[maxn];
void add(int x,int val)
{
	for(int i=x;i<=n;i+=lowbit(i))
		c[i]+=val;
}
int getsum(int x)
{
	int sum=0;
	for(int i=x;i>=1;i-=lowbit(i))
		sum+=c[i];
	return sum;
}
区间修改 + 单点查询

对于区间修改,c [ i ] 依据 dis [ i ] = a [ i ] - a [ i - 1 ] 差分数组进行更新( 如果学过差分,那么对区间加减操作,可以在原数组的差分数组上操作 )。
单点查询,因为差分数组的前缀和就是原数组,所有对差分数组 getsum 就是原数组。

基本代码
int n,m,a[maxn],c[maxn];
void add(int x,int val)
{
	for(int i=x;i<=n;i+=lowbit(i))
		c[i]+=val;
}
int getsum(int x)
{
	int sum=0;
	for(int i=x;i>=1;i-=lowbit(i))
		sum+=c[i];
	return sum;
}
区间修改 + 区间查询
基本代码
int n,q,a[maxn],c1[maxn],c2[maxn];
void add(int x,int val)
{
	for(int i=x;i<=n;i+=lowbit(i))
		c1[i]+=val,c2[i]+=val*(x-1);
}
int getsum(int x)
{
	int sum=0;
	for(int i=x;i>=1;i-=lowbit(i))
		sum+=x*c1[i]-c2[i];
	return sum;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值