POJ1195 Mobile phones 二维树状数组

又是长期没刷题的咸鱼，，今天做了道树状数组的题。

这道题一开始我没想用二维的，想着用一维的树状数组，把二维下标重新排列成一维的，比如(x*n+y+1)。但后来发现这样不行，会多算：比如4×4的矩阵，下标0≤x≤3，0≤y≤3，对应一维下标从1对应到16

           1   2   3   4

           5   6   7   8

           9  10 11 12

          13 14 15 16

若查找范围是(1,1)<->6 到(2,3)<->12 这个矩阵范围，此种方法会把"9"位置上的数也算上。

所以用二维数组。AC代码如下：

#include
#include
#include
#include
using namespace std;
const int INF=0x3f3f3f3f;
#define ll long long

const int MAX=1030;
int n;
ll c[MAX][MAX];
int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int y,int val)
{
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=n;j+=lowbit(j))
            c[i][j]+=val;
}
ll quary(int x,int y)
{
    ll ans=0;
    for(int i=x;i>0;i-=lowbit(i))
        for(int j=y;j>0;j-=lowbit(j))
            ans+=c[i][j];
    return ans;
}

int main()
{
    int ins;
    while(scanf("%d",&ins)==1)
    {
        if(ins==3)
            break;
        else if(ins==0)
        {
            scanf("%d",&n);
            memset(c,0,sizeof(c));
            continue;
        }
        else if(ins==1)
        {
            int x,y,t;
            scanf("%d%d%d",&x,&y,&t);
            x++;y++;
            update(x,y,t);
        }
        else if(ins==2)
        {
            int l,b,r,t;
            scanf("%d%d%d%d",&l,&b,&r,&t);
            l++;b++;r++;t++;
            printf("%d\n",quary(r,t)-quary(r,b-1)-quary(l-1,t)+quary(l-1,b-1));//注意!!尤其是l-1,b-1
        }
    }
	return 0;
}