【BZOJ】1452[JSOI2009]Count-优快云博客

本文链接：https://blog.youkuaiyun.com/NLt_XXy/article/details/79587104

本文介绍了一种使用二维树状数组解决特定类型问题的方法。针对输入的小范围n、m和c，通过存储不同颜色的二维树状数组，实现快速查询和更新。适用于处理二维平面上的颜色变化和区域颜色计数等问题。

Description

Input

Output

Sample Input

Sample Output

1
2

HINT

题解：

题目中n,m,c的范围都很小，我们考虑用二维树状数组来做，对于每一种颜色我们都存一下，最后求和就可以了

代码:

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#define lowbit(x) x&(-x)
using namespace std;
int c[102][304][304],a[304][304],n,m,Q;
void Change(int now,int x,int y,int k)
{
	for(int i=x;i<=n;i+=lowbit(i))
	{
		for(int j=y;j<=m;j+=lowbit(j))
		{
			c[now][i][j]+=k;
		}
	}
}
int Query(int now,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[now][i][j];
		}
	}
	return sum;
}
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]);
			Change(a[i][j],i,j,1);
		}
	}
	scanf("%d",&Q);
	while(Q--)
	{
		int opt,x,y,z,x1,x2,y1,y2,d;
		scanf("%d",&opt);
		if(opt==1)
		{
			scanf("%d%d%d",&x,&y,&z);
			Change(a[x][y],x,y,-1);
			a[x][y]=z;
			Change(a[x][y],x,y,1);
		}
		if(opt==2)
		{
			scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&d);
			printf("%d\n",Query(d,x2,y2)+Query(d,x1-1,y1-1)-Query(d,x2,y1-1)-Query(d,x1-1,y2));
		}
	}
	return 0;
}
/*
3 3
1 2 3
3 2 1
2 1 3
3 
2 1 2 1 2 1
1 2 3 2
2 2 3 2 3 2
*/