[BZOJ]2351 Matrix

2351: [BeiJing2011]Matrix

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 896  Solved: 280
[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

HINT

对于100%的实际测试数据，M、N ≤ 1000，Q = 1000

对于40%的数据，A = 1。

对于80%的数据，A ≤ 10。

对于100%的数据，A ≤ 100。

Source

[Submit][Status][Discuss]

HOME Back

二维哈希即可.

#include<stdio.h>#include<algorithm>#include<vector>using namespace std;const int maxn=1005;const int mod=10007;const int base1=9999987;const int base2=1000007;vector<int> G[mod];int hh,q,n,m,a,b,maxx;unsigned int sum[maxn][maxn],pow1[maxn],pow2[maxn];inline void insert(unsigned int x){    hh=((x>>3)+(x>>7)+x)%mod;     G[hh].push_back(x);}inline bool check(unsigned int x){    hh=((x>>3)+(x>>7)+x)%mod;    for(int i=0;i<G[hh].size();i++)     if(G[hh][i]==x) return true;    return false;}int main(){    scanf("%d%d%d%d",&m,&n,&a,&b),maxx=max(m,n),pow1[0]=pow2[0]=1;    for(register int i=1;i<=m;i++) for(register int j=1;j<=n;j++) scanf("%1d",&sum[i][j]);    for(register int i=1;i<=m;i++) for(register int j=1;j<=n;j++) sum[i][j]+=sum[i-1][j]*base1;    for(register int i=1;i<=m;i++) for(register int j=1;j<=n;j++) sum[i][j]+=sum[i][j-1]*base2;    for(register int i=1;i<=maxx;i++)  pow1[i]=pow1[i-1]*base1,pow2[i]=pow2[i-1]*base2;    for(register int i=a;i<=m;i++)     for(register int j=b;j<=n;j++){         unsigned int seed=sum[i][j]-sum[i-a][j]*pow1[a]-sum[i][j-b]*pow2[b]+sum[i-a][j-b]*pow1[a]*pow2[b];         insert(seed);}scanf("%d",&q);while(q--){   for(register int i=1;i<=a;i++) for(register int j=1;j<=b;j++) scanf("%1d",&sum[i][j]);   for(register int i=1;i<=a;i++) for(register int j=1;j<=b;j++) sum[i][j]+=sum[i-1][j]*base1;   for(register int i=1;i<=a;i++) for(register int j=1;j<=b;j++) sum[i][j]+=sum[i][j-1]*base2;   puts(check(sum[a][b])?"1":"0");}}