C语言树状数组的实例详解
最近学了树状数组，给我的感觉就是 这个数据结构好神奇啊^_^
首先她的常数比线段树小，其次她的实现复杂度也远低于线段树 （并没有黑线段树的意思=-=）
所以熟练掌握她是非常有必要的。。
关于树状数组的基础知识与原理网上一搜一大堆，我就不赘述了，就谈一些树状数组的应用好了
1，单点修改，求区间和
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
#define lowbit(x) (x&-x) // 设 x 的末尾零的个数为 y , 则 lowbit(x) == 2^y
void Update(int i,int v) // 初始化与单点修改
{
while(i <= n)
{
c[i] += v ;
i += lowbit(i) ;
}
}
inline int Sum(int i) // 区间求和
{
int res = 0 ;
while(i > 0)
{
res += c[i] ;
i -= lowbit(i) ;
}
return res ;
}
|
2，区间修改，单点查询
这里要用到差分的思想
创建一个差分数组c[]，令c[i] = a[i] - a[i-1] （a[i] 表示原本的第i个数）
则a[i] = ( a[i] - a[i-1] ) + ( a[i-1] - a[i-2] ) + ...... + ( a[2] - a[1] ) +a[1]
= c[i] + c[i-1] + ...... + c[2] + c[1]
所以单点查询变成了区间求和
那么区间修改怎么办呢 ？
我们看这样一个例子：
a 1 3 4 5 7 10
c 1 2 1 1 2 3
若我们令区间[2,4]加2，则
a 1 5 6 7 9 10
c 1 4 1 1 2 1
我们可以发现只有c[2]和c[5]的数值改变了，其实原理也很好想，区间内的前后元素差是不变的，只有（区间第一个元素与前一个元素的差） 和 （区间后第一个元素与区间末尾元素的差） 改变了。所以区间修改问题变成了单点修改问题。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
for(int i=1;i<=n;i++)
{
a[i] = read() ;
Update(i,a[i]-a[i-1]);
}
/* int x=0,y=0; // 注释掉的内容是空间上的优化（初学者建议先跳过）
for(int i=1;i<=n;i++)
{
if(i%2)
{
x = read() ;
Update(i,x-y);
}
else
{
y = read() ;
Update(i,y-x) ;
}
} */
int ii ;
int k,x,y;
for(int i=1;i<=m;i++)
{
ii = read() ;
if(ii == 1)
{
x = read() ; y = read() ; k = read() ;
Update(x,k);
Update(y+1,-k);
}
if(ii == 2)
{
x = read() ;
printf("%d\n",Sum(x));
}
}
|
（洛谷有对应的模板题 P3374 与 P3368）
上述就是树状数组最基础的两个应用，日后更深入的学习后再来更新。
如有疑问请留言或到本站社区交流讨论，感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！
原文链接：http://www.cnblogs.com/zub23333/p/7670436.html