HDU 1166 树状数组
树状数组是解决区间查询问题的一种数据结构。
区间查询最简单的形式就是求一段连续区间的和,这种问题可以用前缀和来解决,建立前缀和数组的时间复杂度为O(n),查询的时间复杂度为O(1)。
但是当我们频繁对区间的一些点进行修改时,前缀和数组在每一次修改后的查询前都要更新,每次更新都是O(n),这样假如修改并查询m次,总的时间复杂度就来到了O(mn)。
树状数组可以降低复杂度,它的建立时间复杂度为O(nlogn),每次修改的时间复杂度为O(logn),每次查询的时间复杂度也是O(logn)。假如修改并查询n次,那么其总的时间复杂度为O(mlogn)。
树状数组的定义如下:
树状数组C[i]的值由原数组A来决定,C[i] = A[i - 2^k+1] + A[i - 2^k+2] + … + A[i],其中k为i的二进制中后导0的个数,2^k = i&(-i)。
树状数组的基本操作有两个,查询和更新,其中建立可以通过更新来实现。
在查询时,原数组前i项的和sum[i] = C[i] + C[i-2^k1] + C[(i-2k1)-2k2] + …,其中k1为i中后导0的个数,k2为i-2^k1中后导0的个数…
在更新时,将所有与A[i]相关的C都施以与A[i]同样的变化,而与A[i]相关的C有C[i] 、 C[i+2^k1] 、 C[(i+2k1)+2k2] …
在创建时,每次针对当前的A[i]对C进行更新即可,A与C均初始化为0。
代码:
#include <iostream>
#include <string>
#include <string.h>
#include <cstdio>
using namespace std;
const int maxn = 50005;
int A[maxn],C[maxn];
int n = 0;
// 计算i对应的2^k,即lowbit
int lowbit(int i){
return i&(-i);
}
// 单点更新,同时也可以用作建立树状数组
void _update(int i,int k){
while(i <= n){
C[i] += k;
i += lowbit(i);
}
}
// 区间查询
int _getSum(int i){
int sum = 0;
while(i > 0){
sum += C[i];
i -= lowbit(i);
}
return sum;
}
int main(){
int T;
cin>>T;
int t = 0;
while(T--){
memset(A, 0, sizeof(A));
memset(C, 0, sizeof(C));
cin>>n;
for(int i = 1;i <= n;i++){
scanf("%d",&A[i]);
_update(i, A[i]);
}
cout<<"Case "<<++t<<":"<<endl;
string s;
int i,j;
while(cin>>s && s[0] != 'E'){
scanf("%d%d",&i,&j);
if(s[0] == 'Q'){
cout<<_getSum(j)-_getSum(i-1)<<endl;
}else if(s[0] == 'A'){
_update(i, j);
}else if(s[0] == 'S'){
_update(i, -j);
}
}
}
}
577ms 2204kB