数据结构:树状数组
声明:全部代码未经编译,不保证正确性,仅限逻辑学习,请勿直接抄袭
什么是树状数组
树状数组,本质是运用二进制运算规则维护区间。它的效率高于线段树,空间也少于线段树。但是所能维护的数据操作比较单一。
lowbit函数
lowbit(x)
,作用是得到
写法如下:
int lowbit(int x){
return x-(x&(x-1));
}
或
int lowbit(int x){
return x&-x;
}
为何如此,自己推一下就行了。
树状数组构建
严格来说树状数组不需提前构建,它只有一个add(x,k)
函数,用于在数组x位置加上或更改为k。
设原数组a[n]
,树状数组tree[n]
,则tree[i]
代表在原数组位置 lowbit(i)
个数(包括第
单点修改代码:
void add(int x,int k){
while(x<=n){
tree[x]+=k;
x+=lowbit(x);
}
}
树状数组区间求和
由于树状数组维护的是部分前缀和,所以很简单。
代码:
int sum(int x){
int ans=0;
while(x>=1){
ans+=tree[x];
x-=lowbit(x);
}
return ans;
}
//区间[l,r]的和就是sum(r)-sum(l-1);
树状数组区间修改
这里用到了差分的思想。
原数列:0 0 0 0 0
我们要给2-4那一段加上x,可以这样操作:
0 x 0 0 -x
然后进行单点查询时加上原数组中的数就行了。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 单线程的Redis速度为什么快?
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 展开说说关于C#中ORM框架的用法!