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 intmap<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");elseprintf("0\n");}return 0;}