情书密码[ 树状数组 ]

情书密码[ 树状数组 ]

题目描述

有消息称：绝恋找到了自己的Miss Right，正准备自己的表白。绝恋已经写好了情书，但为了避免其它人截获，他对情书进行加密。 为了打探绝恋的私密，你冒着生命危险终于搞到了这封情书。原以为可以轻易将情书解密，结果竟然发现聪明的绝恋并没有直接写出加密用的密码，而是在那粉红色的信纸背面写着“T=你的幸运数字”。 就这么放弃了吗？不，作为一个高智商的**，你决不轻言放弃。你还搞到了绝恋做密码时的草稿，通过一定的分析，你发现草稿中隐藏了绝恋的密码，具体规则如下： 草稿纸上写着一个$N\ast M$ 的矩阵，每个位置都有一个数字$C$，绝恋对该矩阵不断进行操作，有时会修改某一位置的值，还不时计算出一个矩形内特定数字的个数作为密码的一部分，绝恋十分聪明，他进行了许多次这样的操作，因此密码也异常复杂，但是你已经下定决心要算出密码了，所以你一定要算出来！！

题意概括
$N\ast M$的矩阵中，两种操作：
①改变一个数字；
②查询某范围内某数字的个数；
真无语，为什么要看别人的情书？？自己不会解密还找我？？

输入格式

第一行有两个数字$N,M$ 接下来$N$ 行，每行$M$ 个数，第$i+1$ 行第$j$个数表示格子$（i,j）$的初始值。 接下来一个整数$Q$ ,接下来$Q$ 行，每行描述一个操作,操作$1：“1XYC”$(不含引号),表示将格子$(X,Y)$的权值改为$C$ 操作$2：“2X1X2Y1Y2C”$(不含引号)，表示询问矩形内有多少个位置的权值为$C$。$X1,X2$ 分别为矩形横坐标中的最小值和最大值，$Y1,Y2$ 为矩形纵坐标中的最小值和最大值。

样例输入

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

输出格式

对于每一个操作2，输出一个整数表示答案，每数一行。

样例输出

1
2

---

思路分析

直接打暴力枚举30分，因此我们考虑更快捷的方法：树状数组
很明显的一个二维树状数组；
不同的是：树状数组求前缀和，这里由于需要查找某个数的个数，因此我们把前缀和改为一个数组，令这个数组的第$i$项表示数字$i$的个数，那么树状数组就成了一个三维的

---

$ACcode$

#include<bits/stdc++.h>
using namespace std;
int m,n,q;
int s[310][310];//存储原数据
int tr[310][310][110];//树状数组

int lowbit(int x)//基本函数
{
	return x&(-x);
}

void add(int x,int y,int past,int now)//添加和替换
{
	for(int i=x;i<=n;i+=lowbit(i))
		for(int j=y;j<=m;j+=lowbit(j))
		{
			tr[i][j][past]--;//一定要记得原数据减一
			tr[i][j][now]++;
		}
}

int sum(int x,int y,int f)//求f的个数
{
	int re=0;
	for(int i=x;i>0;i-=lowbit(i))
		for(int j=y;j>0;j-=lowbit(j))
		{
			re+=tr[i][j][f];
		}
	return re;
}

int find(int x1,int x2,int y1,int y2,int f)
{
	int ans=0;
	ans=sum(x1-1,y1-1,f)+sum(x2,y2,f)-sum(x1-1,y2,f)-sum(x2,y1-1,f);//简单的一个二维前缀和计算
	return ans;
}

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)//初始化
		{
			scanf("%d",&s[i][j]);
			add(i,j,0,s[i][j]);//把0替换s[i][j]
		}
	scanf("%d",&q);
	int k,c;
	for(int i=1;i<=q;i++)
	{
		scanf("%d",&k);
		if(k==1)
		{
			int x,y;
			scanf("%d%d%d",&x,&y,&c);
			int p=s[x][y];
			s[x][y]=c;
			add(x,y,p,c);
		}
		if(k==2)
		{
			int x1,y1,x2,y2;
			scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
			int tot=find(x1,x2,y1,y2,c);
			printf("%d\n",tot);
		}
	}
	return 0;
}

看了这么久，还不点个赞？？