树状数组

树状数组在对数组进行修改和查询时，可在$O(log(n))$的时间内完成。其具体构造过程见博文彻底弄懂二维树状数组

下面是一维树状数组核心的三个函数：

#define MAX 32010
int tree[MAX];      //定义树状数组

int lowbit(int x){      //返回x对应的二进制最右边的第一个1的位置
    return x&(-x);
}

void add(int i, int x){     //增加x时，更新树状数组
    while(i<MAX){
        tree[i] += x;
        i += lowbit(i);
    }
}

int sum(int x){         //求位置x之前包括x的元素之和
    int s=0;
    while(x>0){
        s += tree[x];
        x -= lowbit(x);
    }
    return s;
}

杭电oj-1541应用树状数组代码如下：

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;

#define MAX 32010

int tree[MAX];      //树状数组
int lev[MAX];       //所求级别数组

int lowbit(int x){
    return x&(-x);
}

void add(int x){
    while(x<MAX){
        tree[x]++;
        x += lowbit(x);
    }
}

int sum(int x){
    int s=0;
    while(x>0){
        s += tree[x];
        x -= lowbit(x);
    }
    return s;
}

int main(){
    int N;
    freopen("input.txt","r",stdin);
    while(scanf("%d",&N) != EOF){
        int x,y;
        memset(tree,0,sizeof(tree));
        memset(lev,0,sizeof(lev));
        for(int i=0; i<N; i++){
            scanf("%d%d",&x,&y);
            lev[sum(x+1)]++;
            add(x+1);
        }
        for(int i=0; i<N; i++)
            cout<<lev[i]<<endl;
    }
    return 0;
}

对应的二维树状数组的核心函数如下：

int tree[XMAX][YMAX];       //定义二维树状数组

void add(int i, int j, int x){
    for(int x=i; x<len1; x+=lowbit(x)){      //len1为输入数组的行数
        for(int y=j; y<len2; y+=lowbit(y)){        //len2为输入数组的列数
            tree[x][y] += x;
        }
    }
}

int sum(int i, int j){
    int s=0;
    for(int x=i; x>0; x-=lowbit(x)){
        for(int y=j; y>0; y-=lowbit(y)){
            s += tree[x][y];
        }
    }
    return s;
}