HDU 1166 树状数组

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-2^k1)-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+2^k1)+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