首页 > 编程语言 > 详细

树状数组

时间:2015-04-06 21:45:33      阅读:290      评论:0      收藏:0      [点我收藏+]

序:说起树状数组,记得这是我正式去组里的第一天,一点基础都没有的我顿时就懵了,学长在上面讲,完全不知所云,于是那天晚上就失眠了。。。

正文:

    一 树状数组就是给你一个数组int a[N],对这个数组进行如下操作:

      (1)修改数组中某元素的值;

      (2)查询该数组任意区间的和。

        任何算法都是以减少复杂度为目的的。其复杂度为(M+Q)log2(N);

        (M为修改次数,Q为查询次数,N为数组长度)

技术分享

   

      首先设一个C数组构成一个树状关系:C[x]分布在 log2(lowbit(x)) 层;每个C[k]只有一个父亲节点,子节点与其父节点之间的距离为lowbit(k)。

      对(1)操作,修改某位置k的值,则对于数组C要修改包括a[k]的位置,k与下一个节点的距离为lowbit(k)。

      对(2)操作,查询某位置k的sum(k),则一次查询包括k到1的sum,k与其下一个相关节点的位置的距离为lowbit(k)。

     函数实现代码:

技术分享
int lowbit(int x)   //位运算求lowbit(x)
{
    return x&(-x);
}

int getsum(int x)     //求sum
{
    int result=0;
    while(x>0)
   {
    result +=c[x];
    x-=lowbit(x);
   }
   return result;
} 

void add(int x)      //修改位置为x的值,val为变化值,可为+-;
{
      while(x<=Max)
      {
        c[x]+=val;
        x+=lowbit(x);
       }
}
View Code

     也可构造二维数状数组

技术分享
int lowbit(int x)
{
   return x&(-x);
}

void add(int x,int y,int val)
{
   int i,j;
   for(i=x;i<=n;i+=lowbit(i))
   for(j=y;j<=n;j+=lowbit(j))
   c[i][j]+=val;
}

int getsum(int x,int y)
{
   int i,j,sum=0;

   for(i=x;i>0;i-=lowbit(i))
   for(j=y;j>0;j-=lowbit(j))
   sum+=c[i][j];
   return sum;
}
View Code

   二 应用

    (1)假如给定一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[1],a[2]...a[i-1]中(即位 置i的左边)小于等于a[i]的数的个数。对此例b[] = {0,1,1,2,0}。 那么该如何去求得b[i]呢?

    (2)假如给你一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[1],a[2]...a[i-1]中(即位置i的左边)大于等于a[i]的数的个数。对此例b[] = {0,0,1,1,4}。 那么该如何去求得b[i]呢?

    (3)假如给你一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[i],a[i+1]...a[n]中(即位置i的右边)小于等于a[i]的数的个数。对此例b[] = {1,3,1,1,0}。 那么该如何去求得b[i]呢?

    (4)假如给你一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[i],a[i+1]...a[n]中(即位置i的右边)大于等于a[i]的数的个数。对此例b[] = {3,0,1,0,0}。 那么该如何去求得b[i]呢?

      以上四种情况殊途同归,其中(2)可以将其编号倒过来用(1)的方法求;

                                          (4)可以将其编号倒过来用(3)的方法求。

       用数状数组函数调用方式

技术分享
for(int i=1; i<=n; i++)
{
     b[i] = getSum(a[i]); //求前a[i]项的和
    update(a[i],1);      //第a[i]个元素+1
}
View Code

     虽然现在稍懂了一点,但是还是应用不好,不过吾将上下而求索。

树状数组

原文:http://www.cnblogs.com/searchmushan/p/4396560.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!