首页 > 代码库 > 二维树状数组(水题) POJ1195

二维树状数组(水题) POJ1195

前段时间遇到线段树过不了,树状数组却过了的题。(其实线段树过得了的)

回忆了下树状数组。

  主要原理,还是二进制位数,每一项的和表示其为它的前((最后一位1及其后)的二进制数)和,可从二进制图来看。(用线段树想一想其实只是线段树编号不同而已,本质类似)

写了下二维树状数组,几乎和一维相同,也没必要不同。

#include <cstdio>#include <cstring>int l,r,x,y,n,a,p,sum[1125][1125];inline int lowbit(int x){    return x&(-x);}void update(int x, int y, int p){    for(int i=x; i<=n; i+=lowbit(i))        for(int j=y; j<=n; j+=lowbit(j))        sum[i][j]+=p;}int query(int x, int y){    int ret=0;    for(int i=x; i>0; i-=lowbit(i))        for(int j=y; j>0; j-=lowbit(j))        ret+=sum[i][j];    return ret;}int main(){    while(scanf("%d",&p)!=EOF)    {        if(p==3) break;        switch (p) {            case 0:                memset(sum,0,sizeof(sum));                scanf("%d",&n);                break;            case 1:                scanf("%d%d%d",&x,&y,&a);                update(x+1,y+1,a);                break;            case 2:                scanf("%d%d%d%d",&x,&y,&l,&r);                printf("%d\n",query(l+1,r+1)-query(l+1,y)-query(x,r+1)+query(x,y));                break;        }    }    return 0;}
View Code

 

二维树状数组(水题) POJ1195