POJ 1195 Mobile phones(裸的二维树状数组）-优快云博客

本文介绍了一个使用二维树状数组解决的POJ 1195题目。该算法可以在O(logN)的时间复杂度内更新矩阵中某一点的值，并求出任意子矩阵的元素之和。文章提供了完整的C++实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给出一个矩阵，给某个格子加/减一个数，就某个子矩阵的和，1024*1024的范围，二维的树状数组

子矩阵(x1,y1,x2,y2)（左下角的点和右上角的点）的和ans=sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1)

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#define nMAX 1050
using namespace std;
int c[nMAX][nMAX],n;
int lowbit(int x)
{
    return x&(-x);
}
void add(int i,int j,int w)
{
    int temp;
    while(i<=n)
    {
        temp=j;
        while(temp<=n)
        {
            c[i][temp]+=w;
            temp+=lowbit(temp);
        }
        i+=lowbit(i);
    }
}
int sum(int i,int j)//与一维的稍有不同
{
    int ans=0,temp;
    while(i>=1)
    {
        temp=j;
        while(temp>=1)
        {
            ans+=c[i][temp];
            temp-=lowbit(temp);
        }
        i-=lowbit(i);
    }
    return ans;
}
int main()
{
    int i,j,k,f,commend;
    while(~scanf("%d",&commend))
    {
       if(commend==3)break;
       if(commend==0)
       {
           scanf("%d",&n);
           memset(c,0,sizeof(c));
       }
       if(commend==1)
       {
           scanf("%d%d%d",&i,&j,&k);//下表必须从1,1开始，因为lowbit(0)=0死循环
           add(i+1,j+1,k);
       }
       if(commend==2)
       {
           scanf("%d%d%d%d",&i,&j,&k,&f);
           i++,j++,k++,f++;
           int ans=sum(k,f)-sum(i-1,f)-sum(k,j-1)+sum(i-1,j-1);
           printf("%d\n",ans);
       }
    }
    return 0;
}

转载于:https://www.cnblogs.com/sdau10kuaile/archive/2012/08/17/2644700.html