[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");
	}
}