Phantom

努力~~~

POJ 1195 二维树状数组

二维树状数组的模版题。写这个题的时候注意不要把a数组直接理解为普通的二维矩阵,了因为i和j的累加和累减方式不同,所以实际上不是一格格的跑的。应该按照一维的树状数组的推广来理解即可。

#include<cstdio>
#include<cstring>
#include<algorithm>


using namespace std;

const int maxn=2000;
int num,n,x,y,z,h;
int a[maxn][maxn];

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

void add(int x,int y,int z)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        for(int j=y;j<=n;j+=lowbit(j))
        {
            a[i][j]+=z;
        }
    }
}

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

}

int main()
{
    while(scanf("%d",&num)!=EOF)
    {
        if(num==3)
            break;
        else if(num==0)
        {
            scanf("%d",&n);
            memset(a,0,sizeof(a));
        }
        else if(num==1)
        {
            scanf("%d%d%d",&x,&y,&z);
            add(x+1,y+1,z);
        }
        else if(num==2)
        {
           scanf("%d%d%d%d",&x,&y,&z,&h);
           int ans=0;
            ans=sum(z+1,h+1)-sum(z+1,y)-sum(x,h+1)+sum(x,y);
           printf("%d\n",ans);
        }
    }
    return 0;
}


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_23702679/article/details/52356019
个人分类: 树状数组
上一篇POJ 2352 树状数组
下一篇POJ 2155 二维数组
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭