题目传送门:https://www.luogu.org/problemnew/show/P4054
题意:
有一个n*m的矩阵,矩阵中每一个各自有不同的颜色。现在有q中操作,[1]将(x,y)的颜色改为z;[2]询问(x1,y1),(x2,y2)中颜色为z的颜色的个数。
思路:
二维树状数组(其实是三维,但核心思想就是二维)。
c[z][x][y]=1表示在(x,y)位置上有一个颜色为z的,c[z][x][y]=0表示在(x,y)位置上没有颜色为z的。
再用一个color[x][y]表示(x,y)位置的颜色。
然后更新color数组,维护c数组即可。
代码:
#include<cstdio> #define lowbit(x) ((x)&(-(x))) int n,m,q; int color[310][310],c[110][310][310]; void add(int col,int x,int y,int z) { for(int i=x;i<=n;i+=lowbit(i)) for(int j=y;j<=m;j+=lowbit(j)) c[col][i][j]+=z; } int getsum(int col,int x,int y) { int sum=0; for(int i=x;i;i-=lowbit(i)) for(int j=y;j;j-=lowbit(j)) sum+=c[col][i][j]; return sum; } int main() { int t,x1,x2,y1,y2,x,y,z; scanf("%d %d",&n,&m); for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) { scanf("%d",&color[i][j]); add(color[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,&z); add(color[x][y],x,y,-1); color[x][y]=z; add(color[x][y],x,y,1); } else { scanf("%d %d %d %d %d",&x1,&x2,&y1,&y2,&z); printf("%d\n",getsum(z,x2,y2)-getsum(z,x1-1,y2)-getsum(z,x2,y1-1)+getsum(z,x1-1,y1-1)); } } }