给定一个 M 行 N 列的 01 矩阵(只包含数字 0 或 1 的矩阵),再执行 Q 次询问,每次询问给出一个 A 行 B 列的 01 矩阵,求该矩阵是否在原矩阵中出现过。
输入格式
第一行四个整数 M,N,A,B。
接下来一个 M 行 N 列的 01 矩阵,数字之间没有空格。
接下来一个整数 Q。
接下来 Q 个 A 行 B 列的 01 矩阵,数字之间没有空格。
输出格式
对于每个询问,输出 1 表示出现过,0 表示没有出现过。
数据范围
A≤100,M,N,B≤1000,Q≤1000
输入样例:
3 3 2 2
111
000
111
3
11
00
11
11
00
11
输出样例:
1
0
1
思路:字符串哈希
暴力解法:暴力枚举原矩阵每个a*b的矩阵,比较是否相等,毫无疑问会超时
优化:将原矩阵每个a*b的矩阵用字符串哈希的一个值来代替该a*b矩阵,然后将所有哈希值存在一起,
每次输入的矩阵也用哈希值来代替,从而在原矩阵的哈希值里面查找是否有相同的值就可以了;
注意:二维哈希与一维哈希有相似之处,
对于一个矩阵:
111
000
111
进行二维哈希可以将每一行看成一维哈希里面的一个p进制数,从而利用一维哈希得出二维哈希
#include<iostream>
#include<unordered_set>
using namespace std;
using ull=unsigned long long;
const int N=1e3+10,P=131;
char str[N];
ull f[N][N],p[N*N];
ull get(ull f[],int l,int r)//一维哈希求l,r区间哈希值
{
return f[r]-f[l-1]*p[r-l+1];
}
int main()
{
int n,m,a,b;
cin>>n>>m>>a>>b;
p[0]=1;
for(int i=1;i<=m*n;++i) p[i]=p[i-1]*P;//求p进制的i次方
for(int i=1;i<=n;++i){
cin>>str+1;
for(int j=1;j<=m;++j)
f[i][j]=f[i][j-1]*P+str[j]-'0';//将每一行进行哈希
}
unordered_set<ull> hash;
for(int i=b;i<=m;++i)//枚举每一个a*b的矩阵进行哈希,先枚举列,再枚举行
{
ull res=0;
int l=i-b+1,r=i;//枚举每一行左右端点
for(int j=1;j<=n;j++)
{
res=res*p[b]+get(f[j],l,r);//将每一行看成一个数相当于一维哈希
if(j>a) res-=get(f[j-a],l,r)*p[a*b];//求a*b的矩阵,行数>a的需要将最上面那一行减去
if(j>=a) hash.insert(res);//插入哈希值
}
}
int k;
cin>>k;
while(k--){
ull res=0;
for(int i=0;i<a;++i){
cin>>str;
for(int j=0;j<b;++j)
res=res*P+str[j]-'0';//求矩阵哈希值
}
if(hash.count(res)) puts("1");
else puts("0");
}
return 0;
}