树状数组与线段树类似,一般用于某些点的操作及区间求和。。时间复杂度也小。
树状数组的基本形状:
由图可知:
。每列的顶端为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;
}