bzoj 2351: [BeiJing2011]Matrix（二维Hash）

2351: [BeiJing2011]Matrix

Time Limit: 20 Sec   Memory Limit: 128 MB
Submit: 938   Solved: 303
[ Submit][ Status][ Discuss]

Description

给定一个M行N列的01矩阵，以及Q个A行B列的01矩阵，你需要求出这Q个矩阵哪些在原矩阵中出现过。
所谓01矩阵，就是矩阵中所有元素不是0就是1。

Input

输入文件的第一行为M、N、A、B，参见题目描述。
接下来M行，每行N个字符，非0即1，描述原矩阵。
接下来一行为你要处理的询问数Q。
接下来Q个矩阵，一共Q*A行，每行B个字符，描述Q个01矩阵。

Output

你需要输出Q行，每行为0或者1，表示这个矩阵是否出现过，0表示没有出现过，1表示出现过。

Sample Input

3 3 2 2

111

000

111

3

11

00

11

11

00

11

Sample Output

1

0

1

很水的二维Hash

每次询问a*b都是不变的，就直接存下所有a*b子矩阵的Hash值

#include<stdio.h>
#include<map>
using namespace std;
#define x 937728547
#define y 185827533
#define uint unsigned int
map<uint, int> p;
uint k[1005][1005], temp[1005][1005], px[1005]  = {1}, py[1005] = {1};
int main(void)
{
	int n, m, a, b, i, j, q;
	for(i=1;i<=1005;i++)
	{
		px[i] = px[i-1]*x;
		py[i] = py[i-1]*y;
	}
	scanf("%d%d%d%d", &n, &m, &a, &b);
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=m;j++)
			scanf("%1u", &k[i][j]);
	}
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=m;j++)
			k[i][j] += k[i][j-1]*x;
	}
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=m;j++)
		{
			k[i][j] += k[i-1][j]*y;
			if(i>=a && j>=b)
				p[k[i][j]-k[i-a][j]*py[a]-k[i][j-b]*px[b]+k[i-a][j-b]*py[a]*px[b]] = 1;
		}
	}
	scanf("%d", &q);
	while(q--)
	{
		for(i=1;i<=a;i++)
		{
			for(j=1;j<=b;j++)
				scanf("%1u", &temp[i][j]);
		}
		for(i=1;i<=a;i++)
		{
			for(j=1;j<=b;j++)
				temp[i][j] += temp[i][j-1]*x;
		}
		for(i=1;i<=a;i++)
		{
			for(j=1;j<=b;j++)
				temp[i][j] += temp[i-1][j]*y;
		}
		if(p.count(temp[a][b]))
			printf("1\n");
		else
			printf("0\n");
	}
	return 0;
}