【树状数组】树状数组及扩展与应用

一）普通树状数组

树状数组(Binary Indexed Tree(BIT))是一个查询和修改复杂度都为log(n)的数据结构。主要用于查询任意两位之间的所有元素之和，单点修改，单点查询。尽管如此，它的常数很小，虽然线段树的许多操作做不到，它依然有着广泛的应用，在辅以CDQ分治、整体二分时尤为凸显。

这就是树状数组的样子。它的实现：

#define lb(x) (x&-x)
int t[MAXN];
void add(int x,int v)
{
    while(x<=n)
    {
        t[x]+=v;
        x+=lb(x);
    }
}
int query(int x)
{
    int ret=0;
    while(x)
    {
        ret+=t[x];
        x-=lb(x);
    }
    return ret;
}

里面的“lb(x)”本来的作用是找出一个数在二进制下最右端的1的位置的，在这里被用于树状数组的转移。

二）二维树状数组

二维树状数组并不难，就是一维的扩展，也就是两层循环：

#define lb(x) (x&-x)
int t[MAXN][MAXM];
void add(int x,int y,int v)
{
    int tmp=y;
    while(x<=n)
    {
        while(y<=m)
        {
            t[x][y]+=v;
            y+=lb(y); 
        }   
        x+=lb(x);y=tmp;
    }
}
int query(int x,int y)
{
    int ret=0,tmp=y;
    while(x)
    {
        while(y)
        {
            ret+=t[x][y];
            y-=lb(y);
        }
        x-=lb(x);y=tmp;
    }
    return ret;
}

它对于二维的数据适用。
例题：BZOJ1542

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define lb(x) (x&-x)
using namespace std;
int n,m,q,t[105][305][305],mp[305][305];
int op,x,y,x2,y2,c;
void add(int x,int y,int c,int v)
{
    int tmp=y;
    while(x<=n)
    {
        while(y<=m)
        {
            t[c][x][y]+=v;
            y+=lb(y); 
        }   
        x+=lb(x);y=tmp;
    }
}
int query(int x,int y,int c)
{
    int ret=0,tmp=y;
    while(x)
    {
        while(y)
        {
            ret+=t[c][x][y];
            y-=lb(y);
        }
        x-=lb(x);y=tmp;
    }
    return ret;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++) 
    {
        scanf("%d",&mp[i][j]);
        add(i,j,mp[i][j],1);
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d%d%d",&x,&y,&c);
            add(x,y,mp[x][y],-1);
            mp[x][y]=c;
            add(x,y,mp[x][y],1);
        }
        else
        {
            scanf("%d%d%d%d%d",&x,&x2,&y,&y2,&c);
            int ans=query(x2,y2,c)+query(x-1,y-1,c);
            ans-=(query(x-1,y2,c)+query(x2,y-1,c));
            printf("%d\n",ans);
        }
    }
    return 0;
}

三）普通树状数组的区间更新

树状数组总归有缺陷：单点。无论干什么，它只能单点操作，这使它有了极大的局限性。有没有办法改善呢？当然是有的。简单地来说，就是给这个树状数组加一个树状数组，记录它的增量，以实现区间的操作。
具体实现：

#define lb(x) (x&-x)
int t1[MAXN],t2[MAXN];
void add(int *t,int x,int v)
{
    while(x<=n)
    {
        t[x]+=v;
        x+=lb(x);
    }
}
int query(int *t,int x)
{
    int ret=0;
    while(x)
    {
        ret+=t[x];
        x-=lb(x);
    }
    return ret;
}
void add(int l,int r,ll v)
{
    add(t1,l,v),add(t1,r+1,-v);
    add(t2,l,v*(l-1)),add(t2,r+1,-v*r);
}
int query(int l,int r)
{
    return r*query(t1,r)-query(t2,r)
    -(l-1)*query(t1,l-1)+query(t2,l-1);
}

例题：POJ3468

四）二维树状数组的区间更新

原理与上面基本相同：
公式化简下来会出现 5个前缀和，分别为：
原数组基数前缀和，常系数前缀和，仅x系数前缀和，仅y系数前缀和，xy系数前缀和。分别维护即可。