luogu P4054 [JSOI2009]计数问题

题目传送门：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数组即可。

注意：上面的c与代码中的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));
		}
	}
}