我们可以发现N,M,C都十分的小,那么只要开100个二维BIT来维护就可以了。每种颜色对应一个BIT。
然后查询的时候再用二维前缀和来搞搞就可以了。
开始在BZOJ时候是T了的,后来才发现是BIT开太大了,导致寻址过慢。
#include <bits/stdc++.h>
const int max_n=300+5;
int N,M,Q;
int A[max_n][max_n],bit[101][max_n][max_n];
inline int read()
{
register int x=0;
register char ch=getchar();
while(!isdigit(ch))
ch=getchar();
while(isdigit(ch))
{
x=(x<<1)+(x<<3)+ch-‘0‘;
ch=getchar();
}
return x;
}
void modify(int c,int x,int y,int num)
{
for(int i=x;i<=N;i+=i&-i)
for(int j=y;j<=M;j+=j&-j)
bit[c][i][j]+=num;
}
int query(int c,int x,int y)
{
int res=0;
for(int i=x;i;i-=i&-i)
for(int j=y;j;j-=j&-j)
res+=bit[c][i][j];
return res;
}
int main()
{
int opt,x,y,xt,yt,c;
N=read(),M=read();
for(int i=1;i<=N;++i)
{
for(int j=1;j<=M;++j)
{
A[i][j]=read();
modify(A[i][j],i,j,1);
}
}
Q=read();
while(Q--)
{
opt=read();
if(opt==1)
{
x=read(),y=read(),c=read();
if(A[x][y]==c) continue;
modify(A[x][y],x,y,-1);
A[x][y]=c;
modify(A[x][y],x,y,1);
}
else
{
x=read(),xt=read(),y=read(),yt=read(),c=read();
printf("%d\n",query(c,xt,yt)-query(c,xt,y-1)-query(c,x-1,yt)+query(c,x-1,y-1));
}
}
return 0;
}