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<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
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;
}