树状数组总结

树状数组总结

问题:已知数组 a[],元素个数为n,现在更改数组中某些元素的值,求更改后a数组中i到j区间内元素的和(1<=i<=j<=n)。

对于这个问题我们当然可以用最朴素的方法来解决,从a[i]一直累加到a[j], 最坏的情况下复杂度为O(n),对于m次更改和访问来说的复杂度就会为O(m*n),当m和n较大时,复杂度就太大导致我们提交题目时就会TLE了,我们可以想一下,在元素发生改变的个数是比较少的,而我们却把所有的元素都进行了一遍计算,这样其实造成了很多无谓的运算,我们就可以把和分成一个个特定的小区间进行存储,从而减少再次运算时无谓的相加。但是如果我们存任意区间的和话,当n特别大的时候,存储量就会特别大,不易更改区间里的内容。我们整理一下思路,现在已经确定了要存储一些特定区间和的想法,但是不知具体该如何存储与操作。

如图;

因为1的二进制为1,所以c[1] = a[1]

因为2的二进制为10,所以才c[2] = a[1] + a[2];

以此类推,,,。

可以发现假设n用二进制表示时,从右数第一次出现1的位置为k;

则c[n] = a[n - k + 1] + a[n - k] + ...+ a[n];

当n是奇数的时候,二进制右边第一位是1,所以c[n] = a[n];

可以使用 

int lowbit(int x)

{

      return x & (-x);

}求 k

在C语言中
& 是 按位与运算符
-x 是x 的补码补码为取反+1
x&(-x)返回x与2^64的最大公约数,即x最多能被n个2整除就返回2^n;如果x是奇数返回1
所以返回值为0,表示x=0
返回值为1,表示x为奇数
返回值为其他数,表示x为x与2^64的最大公约数

在条件语句中可以判断 该数是否为0,为1,能倍2整除?
利用状态位也可以 判断x是否为0

或者直接返回x与2^64的最大公约数

使用
树状数组进行点求和的时候向后更新

void update(int position,int number)
{
    for(int i = position;i <= n;i += lowbit(i))
             c[i] += number;
}

向前求和

int get(int position)
{
        int sum = 0;
        for(int i = position;i >= 1;i -= lowbit(i))
               sum += c[i];
        return sum;
}

例如:HDU 1556

Color the ball

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 6412    Accepted Submission(s): 3385


Problem Description
N个气球排成一排,从左到右依次编号为1,2,3....N.每次给定2个整数a b(a <= b),lele便为骑上他的“小飞鸽"牌电动车从气球a开始到气球b依次给每个气球涂一次颜色。但是N次以后lele已经忘记了第I个气球已经涂过几次颜色了,你能帮他算出每个气球被涂过几次颜色吗?
 
Input
每个测试实例第一行为一个整数N,(N <= 100000).接下来的N行,每行包括2个整数a b(1 <= a <= b <= N)。
当N = 0,输入结束。
 
Output
每个测试实例输出一行,包括N个整数,第I个数代表第I个气球总共被涂色的次数。
 
Sample Input
3 1 1 2 2 3 3 3 1 1 1 2 1 3 0
 
Sample Output
1 1 1 3 2 1
 
Author
8600
 
Source
 
Recommend
LL
代码:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
int map[100010];
int n;
int a,b;
int lowbit(int x)
{
    return x & (-x);
}
void update(int position,int number)
{
    for(int i = position;i <= n;i += lowbit(i))
       map[i] += number;
}
int get(int position)
{
    int sum = 0;
    for(int i = position;i >= 1;i -= lowbit(i))
       sum += map[i];
    return sum;
}
int main()
{
    printf("%d
",lowbit(6));
    while(scanf("%d",&n),n != 0)
    {
        memset(map,0,sizeof(map));
        for(int i = 1;i <= n;i ++)
        {
            scanf("%d%d",&a,&b);
            update(a,1);update(b + 1,-1);
        }
        for(int i = 1;i < n;i ++)
           printf("%d ",get(i));
        printf("%d
",get(n));
    }
    return 0;
}

  

原文地址:https://www.cnblogs.com/GODLIKEING/p/3352763.html