【二维树状数组】bzoj1452 [JSOI2009]Count

权值的种类只有100种，对每种开个二维BIT，然后是经典操作。

#include<cstdio>
using namespace std;
int n,m,q,x1,y1,x2,y2,val,op,a[301][301];
struct BIT_2D
{
    int d[301][301];
    void update(int x,const int &y,const int &V)
      {
        for(;x<=n;x+=(x&(-x)))
          for(int j=y;j<=n;j+=(j&(-j)))
            d[x][j]+=V;
      }
    int getsum(int x,const int &y)
      {
        int res=0;
        for(;x;x-=(x&(-x)))
          for(int j=y;j;j-=(j&(-j)))
            res+=d[x][j];
        return res;
      }
}T[101];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
      for(int j=1;j<=m;++j)
        {
          scanf("%d",&a[i][j]);
          T[a[i][j]].update(i,j,1);
        } scanf("%d",&q);
    for(;q>0;--q)
      {
          scanf("%d",&op);
          if(op==1)
            {
                scanf("%d%d%d",&x1,&y1,&val);
                T[a[x1][y1]].update(x1,y1,-1);
                T[a[x1][y1]=val].update(x1,y1,1);
            }
          else
            {
                scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&val);
                int t1=T[val].getsum(x2,y2);
                int t2=T[val].getsum(x1-1,y2);
                int t3=T[val].getsum(x2,y1-1);
                int t4=T[val].getsum(x1-1,y1-1);
                printf("%d\n",t1-t2-t3+t4);
            }
      }
    return 0;
}