• 博客园logo
  • 会员
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • HarmonyOS
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录
#define wzz
博客园    首页    新随笔    联系   管理    订阅  订阅
一维,二维数组

树状数组总结:假设c[]为树状数组,a[]为原数组,则两者之间存在这么一个关系,c[i]代表的意义是从a[i]开始往前2^k个元素的和 (k为i化成二进制后尾部包含的0的个数),举例来说就是:(括号后面的数字代表几进制) c[1] = a[1]( (1)10 -> (1)2 -> k=0 ) c[2] = a[1] + a[2]( (2)10 -> (10)2 -> k=1 ) c[3] = a[3]( (3)10 -> (11)2 -> k=0 ) c[4] = a[1] + a[2] + a[3] + a[4]( (4)10 -> (100)2 -> k=2 ) ...... 有位运算的性质可以得到:对于i来说,i&(-i) = 2^k,那么树状数组的基本功能函数自己手算一下就能明白了,下面附上模板:

树状数组模板:元素下标均从1开始

(1).一维树状数组:一定要注意数组c的初始化,N是数组的大小,modify的功能是把第n个元素加上delta,而sum的功能则是求从第一个元素开始到第n个元素之和,当然也可以求一段区间上的和,比如我要求第x个元素到第y个元素的和,答案就是sum(y) - sum(x - 1)

int lowbit( int n )

{ return n & (-n);

}

void modify( int n, int delta )

{

     while( n <= N )

    {

          c[n] += delta;

          n += lowbit(n);

    }

}

int sum( int n )

{

        int ret = 0;

        while( n != 0 )

      {

             ret += c[n];

             n -= lowbit(n);

      }

      return ret;

}

(2).二维树状数组:同样不要忘记c的初始化,modify 的功能是改变元素(x, y),sum的功能则是求从元素(1, 1)开始到(x, y)的总和,同样,可以求出任意一个子矩阵内的所有元素之和,即sum(x2, y2) - sum(x1-1, y2) - sum(x2, y1-1) + sum(x1-1, y1-1)

int lowbit( int x )

{

      return x & (-x);

}

void modify( int x, int y, int delta )

{

       int i, j;

      for(i=x; i<=N; i+=lowbit(i))

    {

            for(j=y; j<=N; j+=lowbit(j))

           {

                 c[i][j] += delta;

           }

     }

}

int sum( int x, int y )

{

           int res = 0, i, j;

          for(i=x; i>0; i-=lowbit(i))

        {

                    for(j=y; j>0; j-=lowbit(j))

                  {

                          res += c[i][j];

                  }

          }

    return res;

}

posted on 2013-09-27 21:05  #define wzz  阅读(492)  评论(0)    收藏  举报
刷新页面返回顶部
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3