树状数组

树状数组与线段树类似,一般用于某些点的操作及区间求和。。时间复杂度也小。

树状数组的基本形状:

由图可知:

。每列的顶端为c[i]数组

  • C[1] = A[1]
  • C[2] = C[1] + A[2] = A[1] + A[2]
  • C[3] = A[3]
  • C[4] = C[2] + C[3] +A[4] = A[1] + A[2] + A[3] + A[4]
  • C[5] = A[5]
  • C[6] = C[5] + A[6] = A[5] + A[6]
  • C[7] = A[7] 
  • C[8] = C[4] + C[6] + C[7] + A[8] = A[1] + A[2] + A[3] + A[4] + A[5] + A[6] + A[7] + A[8]

由此我们也可以发现规律:

将C[]数组的结点序号转化为二进制

1=(001)      C[1]=A[1]                                

2=(010)      C[2]=A[1]+A[2];                          

3=(011)      C[3]=A[3];                                       

4=(100)      C[4]=A[1]+A[2]+A[3]+A[4];

5=(101)      C[5]=A[5];

6=(110)      C[6]=A[5]+A[6];

7=(111)      C[7]=A[7];

8=(1000)    C[8]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6]+A[7]+A[8];

c[i]中a[]的个数与二进制中从低位开始连续0的个数有关,如c[8],二进制为1000,它的个数为2的3次方个。

下面引入一个概念:

lowbit(x) 其实就是取出x的最低位1  换言之  lowbit(x)=2^k ,k表示从最低位开始连续0的个数

一般我们都用x&(-x)位运算来表示lowbit(x)。它其实就是为了联系a数组和c数组的关系的。

因此可求c[i]就等于从a[i]开始向左连续lowbit(x)个a数组的和。。

1、数据修改

当一般对a数组内的某些元素进行修改时,相应的对应的c数组的值也会发生改变,如图,若改变a[1]的值,相继的c[1]、c[2]、c[4]、c[8]的值也会相应改变,那如何由修改a[i]值后,知道修改哪个c[]的值呀?

虽然我也没大看懂它的解释,但我知道了,一般a[k]改变时,c[k]也会发生改变,k +=lowbit(k)。

因此,当对进行a数组修改操作,别忘了修改某些c数组的值呀!

2.求和,区间求和

由上可以推出:

sum[1]=A[1]=C[1]

sum[2]=A[1]+A[2]=C[2]

.............

sum[6]=A[1]+A[2]+A[3]+A[4]+A[5] +A[6]=C[4]+C[6]

sum[7]=A[1]+A[2]+A[3]+A[4]+A[5] +A[6]+A[7]=C[4]+C[6]+C[7]

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

对于求和,不难发现sum与c数组的关系,跟上面类似,求sum[k],从c[k]开始倒着开始找,k=k-lowbit(k)

求区间和时,如(a,b)之间的和,只许用sum(b)-sum(a-1)即可。

sum与C数组和C数组与A数组都与lowbit有关,需要记住并掌握

例题:

题目描述

如题,已知一个数列,你需要进行下面两种操作:

1.将某一个数加上x

2.求出某区间每一个数的和

输入输出格式

输入格式:

 

第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3个整数,表示一个操作,具体如下:

操作1: 格式:1 x k 含义:将第x个数加上k

操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和

 

输出格式:

 

输出包含若干行整数,即为所有操作2的结果。

 

输入输出样例

输入样例#1: 复制

5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4

输出样例#1: 复制

14
16
#include<iostream>
#include<cstdio>
#include<cstring>
#define p 600000
#define lowbit(x) x&(-x)
using namespace std;
int t[p];
int n,m;
void add(int pos,int num){
    while(pos<=n){//从底往上赋值,更新pos节点的各节点的值,每次向上跳lowbit(x)
        t[pos]+=num;
        pos+=lowbit(pos);
    }
}
long long sum(int pos){
    long long sum=0;
    while(pos){
        sum+=t[pos];//将当前节点的值加入到总和中
        pos-=lowbit(pos);//向下开始寻找
    }
    return sum;
}
int main(){
    scanf("%d%d",&n,&m);
    int num;
    memset(t,0,sizeof(t));
    for(int i=1;i<=n;i++){
        scanf("%d",&num);
        add(i,num);//将num值放入i位置
    }
    for(int i=1;i<=m;i++){
        int s,a,b;
        scanf("%d%d%d",&s,&a,&b);
        if(s==1)
            add(a,b);
        else
            printf("%lld\n",sum(b)-sum(a-1));//查询从(a,b)的区间和,可以表示为求1到b与1到a-1的和之差
    }
    return 0;
}


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值