树状数组(一)

如果给定一个数组A1,A2,A3…An。求从第i个数到第j个数的和(包含第i个和第j个且1<=i<=j<=n)
很简单直接累加,用sum[i]表A1+A2+…+Ai的和。sum[j]-sum[i-1]就是所求,时间复杂度为O(n)。

#include<cstdio>
const int N = 1000;
int sum[N]; //sum[i]表示A1+A2+...+Ai的和
int A[N];   //A[i]表示数列的第i个数
int main(void)
{
    int i, j;
    int n;
    scanf("%d", &n);
    for(i = 1; i <= n; i++)  {scanf("%d", &A[i]); sum[i] = sum[i-1]+A[i];}
    while(~scanf("%d%d", &i, &j))//询问从第i个数到第j个数的和
        printf("%d\n", sum[j] - sum[i-1]);
    return 0;
}

如果给定一个数组A1,A2,A3…An。可能的操作为:

  1. 第i个数加d
  2. 计算从第i个数到第j个数的和(包含第i个和第j个且1<=i<=j<=n)

类似的题目:hdu1166
如果还用到上面的方法,代码如下:

#include<cstdio>
const int N = 1000;
int sum[N]; //sum[i]表示A1+A2+...+Ai的和
int A[N];   //A[i]表示数列的第i个数
int n;

void update(int i, int d)
{
    for(; i <= n; i++)  sum[i] += d;
}

int main(void)
{
    int i, j;
    scanf("%d", &n);
    for(i = 1; i <= n; i++)  {scanf("%d", &A[i]); sum[i] = sum[i-1]+A[i];}
    int m, x, y;
    //m表示第m个操作
    while(~scanf("%d%d%d", &m, &x, &y))
    {
        if(m == 1) update(x, y);//m==1时让第x个数加y
        else       printf("%d\n", sum[y] - sum[x-1]);//m==2时询问从第x个数到第y个数的和
    }
    return 0;
}

如果有m次操作时间复杂度就为O(m*n),更改第i个数时,sum[i]后面的都要跟新,这样就显得太耗时了。有更好的方法吗?
sum[i]表示A1+A2+…+Ai的和,暂且称为sum[i]管辖的范围为1~i,虽然查询很快,但是当第i个数更改时sum[i],sum[i+1]…sum[n]都要更改。sum数组管辖的范围越大更新的可能越大。有没有方法缩小sum数组管辖的范围。
树状数组就很巧妙的解决了这个问题。
说到树状数组就少不了下面这张图
这里写图片描述
定义An表示任意n位二进制数,Bn表示各个位全是0的n位二进制数,Cn表示各个位全是1的n位二进制数
树状数组使用二进制数解决了这个问题。An1Bm(二进制数)所管辖的范围为2^m个数。例如当那个数是8(1000二进制)时它管辖的范围为2^3(包含A1..A8),当那个数是6(110二进制)时它管辖的范围为2^1(包含A5,A6),当那个数是5(101二进制)时它管辖的范围为2^0(包含A5)。
假设x的原码为An1Bm
那么-x的反码为A’n0Cm (A’n为An的反码)
那么-x的补码为A’n1Bm
(-x)&x=1Bm=2^m
所以第x个数管辖的范围为(-x)&x。

比如x的原码为00001010
-x的反码为11110101
-x的补码为11110110
11111010&00001010=10=2^1

c[i]表示i管辖的范围内的数和被管辖的数改变了,管辖的数也要改变。被管辖的数是x时,管辖的数则为x+(-x)&x。(这里说的管辖是直接管辖,所以跟新时管辖点跟新了,管辖点的管辖点也要跟新…)
第x个数到第y个数的和,就是前y项和减前x-1项和
前n项和于前m(n-第n项管辖的范围)项和c[n],前m项和只要用同样的方法就可以了。

#include<cstdio>
const int MAXN = 50001;
int c[MAXN];
int n;
//x个数管辖的范围
int lowbit(int x)
{
    return (-x)&x;
}

//求前end项和
int Sum(int end)
{
    int sum = 0;
    while(end > 0)
    {
        sum += c[end];
        end -= lowbit(end);
    }
    return sum;
}

//第pos个元素加num
void update(int pos, int num)
{
    while(pos <= n)
    {
        c[pos] += num;
        pos += lowbit(pos);
    }
}

int main(void)
{
    int m, x, y;
    int i;
    scanf("%d", &n);
    int val;
    for(i = 1; i <= n; i++)
    {
        scanf("%d", &val);update(i, val);
    }
    while(~scanf("%d%d%d", &m, &x, &y))
    {
        if(m == 2)
        {
            printf("%d\n", Sum(y)-Sum(x-1));
        }
        else
        {
            update(x, y);
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值