码迷,mamicode.com
首页 > 编程语言 > 详细

树状数组(BIT)

时间:2015-03-18 23:14:13      阅读:138      评论:0      收藏:0      [点我收藏+]

标签:

i的二进制的最后一个1可以通过i&(-i)得到,时间复杂度o(logn)。对于W*H的二维BIT只需要建立H个大小为x轴方向元素个数W的BIT,复杂度O(logW+logH)。同样的方法可以扩展到更高维度的情况。

 1 int sum(int i)
 2 {
 3     int s=0;
 4     while(i>0) {
 5         s+=bit[i];
 6         i-=i&(-i);
 7     }   
 8     return s;
 9 }
10 void add(int i,int x)
11 {
12     while(i<=n) {
13         bit[i]+=x;
14         i+=i&(-i);
15     }
16 }
17 
18 int main()
19 {
20     int x,y;
21     scanf("%d%d",&n,&m);
22     for(int i=1;i<=n;i++) {
23         scanf("%d",&a[i]);
24         add(i,a[i]);
25     }
26     while(m--) {
27         scanf("%d",&x);
28         printf("%d\n",sum(x));//前x项和
29     }
30     
31     return 0;
32 }

 

树状数组(BIT)

标签:

原文地址:http://www.cnblogs.com/sxiszero/p/4348870.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!