BZOJ1452 [JSOI2009]Count（二维树状数组）

建立n(color)个二维树状数组即可 

#include<stdio.h>
#include<stdlib.h>
int map[305][305],c[105][305][305];
int n,m;
void xg(int color,int p,int x,int y)
{
	int i,j;
	for(i=x;i<=n;i+=i&(-i))
		for(j=y;j<=m;j+=j&(-j))
			c[color][i][j]+=p;
}
int cx(int color,int x,int y)
{
	int i,j,sum=0;
	for(i=x;i>0;i-=i&(-i))
		for(j=y;j>0;j-=j&(-j))
			sum+=c[color][i][j];
	return sum;
}
int main()
{
	int Q,i,j,opt,x1,y1,x2,y2,c;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
		{
			scanf("%d",&map[i][j]);
			xg(map[i][j],1,i,j);
		}
	scanf("%d",&Q);
	for(;Q>0;Q--)
	{
		scanf("%d",&opt);
		if(opt==1)
		{
			scanf("%d%d%d",&x1,&y1,&c);
			xg(map[x1][y1],-1,x1,y1);
			map[x1][y1]=c;
			xg(map[x1][y1],1,x1,y1);
		}
		else
		{
			scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
			printf("%d\n",cx(c,x2,y2)-cx(c,x1-1,y2)-cx(c,x2,y1-1)+cx(c,x1-1,y1-1));
		}
	}
	return 0;
}