P4054 [JSOI2009] 计数问题

二维树状数组板子， C[color] [x][y]

 

#include <bits/stdc++.h>
using namespace std ;
 const int N =403 ,M =2e5+4;
 
 #define int long long
 int A[N][N],c[101][N][N] ;
 int n,m,ans[M],ansl; 
 
 int lowbit(int x){
     return x&-x; 
 }
 void add(int T,int x,int y,int v){
     for(int i=x;i<=n;i+=lowbit(i))
      for(int j=y;j<=m;j+=lowbit(j))
       c[T][i][j]+=v;    
 }
 int qq(int T,int x,int y){
     int res=0;
     for(int i=x;i;i-=lowbit(i))
      for(int j=y;j;j-=lowbit(j))
       res+=c[T][i][j];
       
     return res;  
 }
 int solve(int x1,int y1,int x2,int y2,int T){
     return 
     qq(T,x1-1,y1-1)+qq(T,x2,y2)-qq(T,x1-1,y2)-qq(T,x2,y1-1);
 }
 signed main(){ 
     int i,j,tes;
     cin>>n>>m;
     for(i=1;i<=n;i++)
      for(j=1;j<=m;j++) 
          cin>>A[i][j],add(A[i][j],i,j,1) ;
          
     cin>>tes;
     int x,y,z;
     while(tes--){
         int op;cin>>op;
         if(op==1){
             cin>>x>>y>>z; 
             add(A[x][y],x,y,-1);add(z,x,y,1);
             A[x][y]=z;
         }
         else{
             int x1,x2,y1,y2;
             cin>>x1>>x2>>y1>>y2>>z;
             ans[++ansl]=solve(x1,y1,x2,y2,z);
         }
     }
     for(i=1;i<=ansl;i++) cout<<ans[i]<<endl;
 }