NYOJ 116 树状数组

初学树状数组。。

 

 

这是线段树的图，由图可知，c[1]=a[1];c2=a[1]+a[2];c[3]=a[3];c[4]=a[1]+a[2]+a[3]+a[4];c[5]=a[5];

c[6]=a[5]+a[6];c[7]=a[7];c[8]=a[1]+a[2]+a[3]+a[4]+a[5]+a[6]+a[7]+a[8];

可以得到这个规律，i为奇数时，c[i]=a[i],这里1,3,5,7便是例子。偶数时这里就跟2的幂有关系了，i的因子中最多有2的多少次幂，例如4,2的2次幂，为4，所以从a[4]数共4个数的值，即a[1],a[2],a[3],a[4],当i=6时，因子数最大为2，所以数2个数，即a[5],a[6],同理i=8时，2的三次幂，等于8，即a[1]+.....a[8]的值.网上查询可知，公式为

c[n]=a(n-a^k+1)+......+a[n]; (其中k为从右往走数0的个数，也即第一个出现1的位置,起点从0开始)

这里可以用一个函数计算得

int lowbit(int x)

{

return x&(-x);

}

这个函数怎样理解呢？

x=1时，x的二进制表示为(假设位数为8) 0000 0001

     -x的二进制表示为           1111 1111

                        --->  0000 0001 =1;

x=6时  x的二进制表示为           0000 0110

     -x的二进制表示为           1111 1010

                        --->  0000 0010 =2;

这里相信你明白了，若不懂负数的二进制表示，可百度查之

这个函数的目的就是求出2^k (k为往右数第一个出现1的位置起点从0开始或者0出现的个数,见上)

接下来就是求和了，

int Sum(int n)

{

   int sum=0;

   while(n>0)

{

    sum+=c[n];

    n=n-lowbit(n);

}

return sum;

}

这里具体该怎样理解呢？我们可以看成求一段路径的和，不同于普通的一维数组按照下标为1一次递增或递减，这里是按照树的结构，每次logn，即求路径(x,x-lowit(x),....1)为下标的和

最后修改某个结点的值

void change(int i,int x)

{

  while(i<=m)           //m个结点的总和

  {

      c[i]=c[i]+x;

      i=i+lowbit(i);

  }

}

如果上面求和理解了，这步应该很快理解了，改变了某个结点的值，那么对应的路径上相应的点也要改变，结束的条件即为到达根结点。

下面来看看这道题

链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=116

1，若采取常规的方法，求解，时间的复杂度为O(n^2);

2,采取树状数组的方法，AC代码：

#include<iostream>
#include<stdio.h>
#include<cstring>
using namespace std;
int c[1000008],m,Q,t;
int lowbit(int x)
{
 return x&(-x);
}
int Sum(int n)
{
 int sum=0;
 while(n>0)
 {
  sum=sum+c[n];
  n=n-lowbit(n);
 }
 return sum;
}
void change(int i,int x)
{
 while(i<=m)
 {
  c[i]=c[i]+x;
  i=i+lowbit(i);
 }
} 
int main()
{
 scanf("%d%d",&m,&Q);
 char key[10];
 int start,end;
 for(int i=1;i<=m;++i)
 {
   scanf("%d",&t);
   change(i,t);
 }
    while(Q--)
    {
  scanf("%s%d%d",key,&start,&end);
     if(key[0]=='Q')
     printf("%d\n",Sum(end)-Sum(start-1));
     else if(key[0]=='A')
     change(start,end);
 }
 return 0;
}

参考资料来源：http://www.cnblogs.com/zhangshu/archive/2011/08/16/2141396.html

                            http://hi.baidu.com/czyuan_acm/item/b14bff6ab6ffd093c5d249db