[BZOJ1452/Luogu4054][JSOI2009]Count计数问题

题目链接：

BZOJ1452

Luogu4054

二维树状数组题（或许可以毒瘤树套树）

注意到值域范围为\(100\)，对每个数开一个二维树状数组，在对应的树状数组上修改查询即可。

时间复杂度 \(O(Qlognlogm)\)

空间复杂度 \(O(nmc)\)

代码：

#include <cstdio>

int n,m,q,a[305][305];
struct Binary_Indexed_Tree
{
	int c[305][305];
	
	inline void Modify(int x,int ys,int v)
	{
		for(;x<=n;x+=x&-x)
			for(int y=ys;y<=m;y+=y&-y)
				c[x][y]+=v;
	}
	
	inline int Query(int x,int ys)
	{
		int Res=0;
		for(;x;x^=x&-x)
			for(int y=ys;y;y^=y&-y)
				Res+=c[x][y];
		return Res;
	}
}Bit[105];

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]);
			Bit[a[i][j]].Modify(i,j,1);
		}
	scanf("%d",&q);
	for(int op,x,y,xs,ys,c;q--;)
		if(scanf("%d",&op),op==1)
		{
			scanf("%d%d%d",&x,&y,&c);
			Bit[a[x][y]].Modify(x,y,-1);
			Bit[a[x][y]=c].Modify(x,y,1);
		}
		else
		{
			scanf("%d%d%d%d%d",&x,&xs,&y,&ys,&c);
			int r1=Bit[c].Query(xs,ys);
			int r2=Bit[c].Query(xs,y-1);
			int r3=Bit[c].Query(x-1,ys);
			int r4=Bit[c].Query(x-1,y-1);
			printf("%d\n",r1-r2-r3+r4);
		}
	return 0;
}