[JSOI2009]计数问题

一个n*m的方格，初始时每个格子有一个整数权值。接下来每次有2种操作：

改变一个格子的权值；

求一个子矩阵中某种特定权值出现的个数。

输入输出格式

输入格式：
第一行有两个数N,M。

接下来N行，每行M个数，第i+1行第j个数表示格子(i,j)的初始权值。

接下来输入一个整数Q。

之后Q行，每行描述一个操作。

操作1：“ x y c”(不含双引号)。表示将格子(x,y)的权值改成c（<=x<=n,<=y<=m,<=c<=)。

操作2：“ x1 x2 y1 y2 c”(不含双引号，x1<=x2,y1<=y2)。表示询问所有满足格子颜色为c，且x1<=x<=x2,y1<=y<=y2的格子(x,y)的个数。

输出格式：
对于每个操作2，按照在输入中出现的顺序，依次输出一行一个整数表示所求得的个数。

　　二维的点修改，区间查询，所以树状数组当然没得跑啦，二维的和一维的写法是一样的，只是传参的时候多传几个，循环的时候是2层循环而已，然后当然是权值树状数组开桶来记录个数啦。有一个需要注意的地方是，这里因为查询的是某个子矩阵中的某个数的个数，所以和一维的有一点不一样，这里需要像计算二维前缀和一样，先加一大块，再减两小块，在加上重复减的那个小小块。我会说我开始就是这里写错了吗

代码：

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 int n,m,t,q,x,y,xx,yy,w,c[][][],a[][];
 void update(int d,int x,int y,int w)
 {    for(int i=x;i<=n;i+=i&-i)
         for(int j=y;j<=m;j+=j&-j)
             c[d][i][j]+=w;
 }
 int query(int d,int x,int y)
 {    int ans=;
     for(int i=x;i>;i-=i&-i)
         for(int j=y;j>;j-=j&-j)
             ans+=c[d][i][j];
     return ans;
 }
 int main()
 {    freopen("count.in","r",stdin);
     freopen("count.out","w",stdout);
     scanf("%d%d",&n,&m);
     memset(c,,sizeof c);
     for(int i=;i<=n;i++)for(int j=;j<=m;j++)
     {    scanf("%d",&a[i][j]);
         update(a[i][j],i,j,);
     }
     scanf("%d",&q);
     for(int i=;i<=q;i++)
     {    scanf("%d",&t);
         if(t==)
         {    scanf("%d%d%d",&x,&y,&w);
             update(a[x][y],x,y,-);
             update(w,x,y,);
             a[x][y]=w;
         }else
         {    scanf("%d%d%d%d%d",&x,&xx,&y,&yy,&w);
             int ans=query(w,xx,yy)-query(w,x-,yy)-query(w,xx,y-)+query(w,x-,y-);
             printf("%d\n",ans);
         }
     }
     return ;
 }