BZOJ1452计数问题——二维树状数组

Description

一个N*M的方格，初始时每个格子有一个整数权值，接下来每次有2个操作:
改变一个格子的权值
求一个子矩阵中某个特定权值出现的个数
Input

每一行有两个数字N，M
接下来N行，每行M个数字。第i+1行第j个数字表示格子(i,j)的初值
接下来输入一个Q，后面Q行每行描述一个操作
操作1:
1 x y c，表示将格子(x,y)的值变为c
操作2:
2 x1 x2 y1 y2 c，表示询问所有满足格子中数字为c的格子数字
(n,m<=300,Q<=5000)
(1<=x<=N,1<=y<=M,1<=c<=100)
(x1<=x<=x2,y1<=y<=y2)
Output

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

Sample Input

3 3

1 2 3

3 2 1

2 1 3

3

2 1 2 1 2 1

1 2 3 2

2 2 3 2 3 2

Sample Output

1

2

---

这道题算是比较基础的二维树状数组题了，直接对每一个颜色建一个二维树状数组。然后直接进行查询即可。

#include<bits/stdc++.h>
#define MAXN 305
using namespace std;
int read(){
    char c;int x;while(c=getchar(),c<'0'||c>'9');x=c-'0';
    while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';return x;
}
int n,m,q,c[MAXN][MAXN],f[105][MAXN][MAXN];
int lowbit(int x){return x&-x;}
void add(int col,int x,int y,int ad){
    for(int i=x;i<=n;i+=lowbit(i))
     for(int j=y;j<=m;j+=lowbit(j)){
        f[col][i][j]+=ad;
     }
}
int getsum(int col,int x,int y){
    int ans=0;
    for(int i=x;i;i-=lowbit(i))
     for(int j=y;j;j-=lowbit(j)){
        ans+=f[col][i][j];
     }
    return ans;
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;i++)
     for(int j=1;j<=m;j++){
        c[i][j]=read();
        add(c[i][j],i,j,1);
     }
    q=read();
    for(int i=1;i<=q;i++){
        int type=read();
        if(type==1){
            int x=read(),y=read(),col=read();
            add(c[x][y],x,y,-1);
            add(col,x,y,1);
            c[x][y]=col;
        }
        else{
            int x=read(),u=read(),y=read(),v=read(),col=read();
            printf("%d\n",getsum(col,u,v)-getsum(col,x-1,v)-getsum(col,u,y-1)+getsum(col,x-1,y-1));
        }
    }
    return 0;
}