树状数组（一维）

今天学习了一天的树状数组，感觉有点小懂，但是理解还不够透彻，大致现在博客上写写心得，下回不会再回过头来看，，，，ORZ

先讲讲一维的树状数组：·

以下有引用到大牛博客：http://blog.csdn.net/int64ago/article/details/7429868

还有一个大牛博客：http://blog.csdn.net/matrix_reloaded/article/details/32101509

一维树状数组解决的大致问题就是：修改某点的值、求某个区间的和，，

引用大牛的一段话：

树状数组是干什么的？
平常我们会遇到一些对数组进行维护查询的操作，比较常见的如，修改某点的值、求某个区间的和，而这两种恰恰是树状数组的强项！当然，数据规模不大的时候，对于修改某点的值是非常容易的，复杂度是O(1)，但是对于求一个区间的和就要扫一遍了，复杂度是O(N)，如果实时的对数组进行M次修改或求和，最坏的情况下复杂度是O(M*N)，当规模增大后这是划不来的！而树状数组干同样的事复杂度却是O(M*lgN)，别小看这个lg，很大的数一lg就很小了，这个学过数学的都知道吧，不需要我说了。申明一下，看下面的文章一定不要急，只需要看懂每一步最后自然就懂了。

二、树状数组怎么干的？
先看两幅图（网上找的，如果雷同，不要大惊小怪～），下面的说明都是基于这两幅图的，左边的叫A图吧，右边的叫B图：

树状数组就是构造这样的树状数组，这里需要注意的是，c数组并不代表数组的和，比如c【8】=a【1】+…….+a【8】,但是c【7】=a【7】,

引用大牛的话：

是不是很像一颗树？对，这就是为什么叫树状数组了～先看A图，a数组就是我们要维护和查询的数组，但是其实我们整个过程中根本用不到a数组，你可以把它当作一个摆设！c数组才是我们全程关心和操纵的重心。先由图来看看c数组的规则，其中c8 = c4+c6+c7+a8，c6 = c5+a6……先不必纠结怎么做到的，我们只要知道c数组的大致规则即可，很容易知道c8表示a1～a8的和，但是c6却是表示a5～a6的和，为什么会产生这样的区别的呢？或者说发明她的人为什么这样区别对待呢？答案是，这样会使操作更简单！看到这相信有些人就有些感觉了，为什么复杂度被lg了呢？可以看到，c8可以看作a1～a8的左半边和+右半边和，而其中左半边和是确定的c4，右半边其实也是同样的规则把a5～a8一分为二……继续下去都是一分为二直到不能分，可以看看B图。怎么样？是不是有点二分的味道了？对，说白了树状数组就是巧妙的利用了二分，她并不神秘，关键是她的巧妙！

此时我们就是写出更新维护树的代码：

void add(int k,int num)  //num就代表点k加上num值
{  
    while(k<=n)  //n就是数组a长度
    {  
        tree[k]+=num;  //tree数组就相当于c数组
        k+=k&-k;  
    }  
}  

有了上面的基础，说求和就比较简单了。比如求0001～0110的和就直接c[0100]+c[0110]，分析方法与上面的恰好逆过来，而且写法也是逆过来的，具体就不累述了：

int read(int k)//1~k的区间和  
{  
    int sum=0;  
    while(k)  
    {  
        sum+=tree[k];  
        k-=k&-k;  
    }  
    return sum;  
}  

一维数组例题：POJ - 2352

题解：

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <vector>
#include <set>
#include <map>
#include <string>
#include <cstring>

using namespace std;
int c[32000+10],ans[32000+10];
void read(int x,int num)
{
    while(x<=32001)
    {
        c[x]+=num;
        x+=x&-x;
    }
}
int sum(int x)//前x坐标的星星数量
{
    int ans=0;
    while(x)
    {
        ans+=c[x];
        x-=x&-x;
    }
    return ans;
}
int main()
{
    int n;
    //cin>>n;
    scanf("%d",&n);
    memset(c,0,sizeof(c));
    memset(ans,0,sizeof(ans));
    for(int i=1;i<=n;i++)
    {
        int x,y;
        //cin>>x>>y;
        scanf("%d%d",&x,&y);
        ++ans[sum(x+1)];
        read(x+1,1);
    }
    for(int i=0;i<n;i++)
        //cout<<ans[i]<<endl;
        printf("%d\n",ans[i]);
    return 0;
}

例题2：UVA-1513

这题的题意就是给你n个数，1-n，然后每次问一个数K的前面有多少个数，然后把k放到第一个数的位置，

思路就是树状数组，可以把数组前面留出一些部分给后面的数过来时用，然后当一个数到第一个数位置的时候，它原来前面的数要加一个数，比如1 2 3 4 5，把3放到第一个数，然后3后面的部分4，5减一，然后3放到第一个数时，把3后面的数都加一，这样就能实现1 2加一，4 5不变了，，至于某个数前面有多少个数，自然就是求和就行，套用树状数组模板就行

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <vector>
#include <set>
#include <map>
#include <string>
#include <cstring>

using namespace std;
int c[200000+10],pos[200000+10];
void read(int x,int num)
{
    while(x<200000+1)
    {
        c[x]+=num;
        x+=x&-x;
    }
}
int sum(int x)
{
    int ans=0;
    while(x)
    {
        ans+=c[x];
        x-=x&-x;
    }
    return ans;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n,m;
        cin>>n>>m;
        int det=100000;
        memset(c,0,sizeof(c));
        memset(pos,0,sizeof(pos));
        for(int i=1; i<=n; i++)
        {
            pos[i]=det+i;
            read(det+i,1);
        }
        for(int i=1; i<=m; i++)
        {
              int x;
              //cin>>x;
              scanf("%d",&x);
              if(i!=1)
              printf(" ");
              printf("%d",sum(pos[x]-1));
              read(pos[x],-1);
              pos[x]=det;
              det--;
              read(pos[x],1);
              //cout<<endl;
              //cout<<"c[pos[x]]="<<c[pos[x]]<<endl;
              //cout<<"sum(1)="<<sum(1)<<endl;
        }
        printf("\n");
    }
    return 0;
}

补充一个题：

今天写java的时候水的一个树状数组，真傻逼，变量没有初始化，wa到绝望，可能是不怎么写树状数组有点事手生

题目就是求矩阵每一行的逆序对，输出逆序对最多的那一行的编号。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <queue>
#define ri(n) scanf("%d",&n)
#define oi(n) printf("%d\n",n)
#define rl(n) scanf("%lld",&n)
#define ol(n) printf("%lld\n",n)
#define rep(i,l,r) for(i=l;i<=r;i++)
#define rep1(i,l,r) for(i=l;i<r;i++)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int epg=10-8;
int a[10000+10];
int c[10000+10];
void add(int x)
{
    while(x<10000+10)
    {
        c[x]++;
        x+=x&-x;
    }
}
int sum(int x)
{
    int ans=0;
    while(x)
    {
        ans+=c[x];
        x-=x&-x;
    }
    return ans;
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    int ans=0,max_=0,flag=1;
    for(int i=1;i<=k;i++)
    {
        memset(c,0,sizeof(c));
        ans=0;
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&a[j]);
            add(a[j]);
            ans+=j-sum(a[j]);
        }
        if(ans>max_)
        {
            max_=ans;
            flag=i;
        }
    }
    printf("%d\n",flag);
    return 0;
}