[JSOI2009]计数问题

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

改变一个格子的权值；

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

输入输出格式

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

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

接下来输入一个整数Q。

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

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

操作2：“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[105][305][305],a[305][305];
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=0;
    for(int i=x;i>0;i-=i&-i)
        for(int j=y;j>0;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,0,sizeof c);
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)
    {    scanf("%d",&a[i][j]);
        update(a[i][j],i,j,1);
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++)
    {    scanf("%d",&t);
        if(t==1)
        {    scanf("%d%d%d",&x,&y,&w);
            update(a[x][y],x,y,-1);
            update(w,x,y,1);
            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-1,yy)-query(w,xx,y-1)+query(w,x-1,y-1);
            printf("%d
",ans);
        }
    }
    return 0;
}