poj 1195 二维树状数组

POJ1195

    其实只要把它的原理弄懂的话，树状数组真的很简单的。关键要把握两个点：1.更新一个a元素的时候。如果a[i]更新了，那么以下几项都需要更新：c[n1],c[n2],c[n3],....,c[nm];其中n1=i,n(i+1)=ni+lowbit(ni);nm+lowbit(nm)必须大于a的元素个数N。2.sum(k)=c[n1]+c[n2]+...+c[nm];其中nm=k，n(i-1)=ni-lowbit(ni);n1-lowbit(n1)必须小于0。这两点弄请楚了，对于树状数组也很容易理解了。呵呵~

   而在这题里面，还有一个要注意的就是求和的时候，注意矩阵就范围的方法。

 好了，上代码吧.

 

#include <iostream>
#include <memory.h>
#define MAXN 1025
using namespace std;

int C[MAXN][MAXN];
int lowbit[MAXN];
int i,j;
int s; //矩阵的维数
void Modify(int x,int y,int a){
    for(i=x ;i<=s ;i+=lowbit[i])
        for(j=y ;j<=s;j+=lowbit[j]){
            C[i][j]+=a;
        }        
}

int getSum(int x,int y){
     //查询第1行到第x行，第1列到第y列的和
    int sum=0;
      for(i=x ;i>0 ;i-=lowbit[i])     
         for(j=y ;j>0 ;j-=lowbit[j])
            sum+=C[i][j];
    return sum;
}

int main(){
    int Ins;
    int X,Y,A;
    int L,B,R,T;
    int flag=1;
    //lowbit
    for(i=1;i<=MAXN ;i++){
       lowbit[i]=i&(i^(i-1));
    }
    while(true){
        scanf("%d",&Ins);
        if(Ins==0){
          scanf("%d",&s);
          memset(C,0,sizeof(C));
        }
        else if(Ins==1){
          scanf("%d%d%d",&X,&Y,&A);
          Modify(X+1,Y+1,A);
       }
       else if(Ins==2){
         scanf("%d%d%d%d",&L,&B,&R,&T);
         L++;B++;R++;T++;
         printf("%d\n",getSum(R,T)+getSum(L-1,B-1)-getSum(R,B-1)-getSum(L-1,T));
       }
       else{
         return 0; 
       }
    }
     
}