UVA 11019 Matrix Matcher（字符串hash+KMP）

题目大意：

     给你两个字符矩阵，然后问A矩阵中有几个B矩阵

解决方法：

   A矩阵x行y列   B矩阵 a行b列

   将A矩阵一列没a行hash一次，hash成一个x-a+1行y列的矩阵

   将B矩阵每一列都hash成一个数值，变成一个1行b列的矩阵

   将B作为模板串对与A的每一行进行KMP ，最后就得到结果了

我的代码：

#include <cstdio>
#include <string>
#include <iostream>
#include <cstring>
#define seed 131
#define ull unsigned long long
using namespace std;
int n,m,x,y;
char s1[1010][1010],s2[1010][1010];
ull hx[1010][1010],zx[1010],base[1010];
int f[1010];
int hashs(){
for (int i=0;i<m;i++){
ull a=0;
int totl=0;
for (int j=0;j<n;j++){
if (j<x){
a=a*seed+s1[j][i];
if (j==x-1) hx[totl++][i]=a;
}
else {
a=a*seed+s1[j][i]-s1[j-x][i]*base[x];
hx[totl++][i]=a;
}
//cout<<j<<" "<<i<<" "<<a<<endl;
}
}
for (int i=0;i<y;i++){
ull a=0;
for (int j=0;j<x;j++){
a=a*seed+s2[j][i];
if (j==x-1) zx[i]=a;
//cout<<j<<" "<<a<<" "<<s2[j][i]<<endl;
}
}
return 0;
}
void getFail(ull *P,int *f){
int m=y;
f[0]=f[1]=0;
for (int i=1;i<m;i++){
int j=f[i];
while (j&&P[i]!=P[j]) j=f[j];
f[i+1]=P[i]==P[j]?j+1:0;
}
}
int kmp(ull *T ,ull *P,int *f){
int ans=0;
int n=m,mm=y;
getFail(P,f);
int j=0;
for (int i=0;i<n;i++){
while (j&&P[j]!=T[i]) j=f[j];
if (P[j]==T[i]) j++;
if (j==mm) ans++;
}
return ans;
}
int main (){
//freopen("test.in","r",stdin);
base[0]=1;for (int i=1;i<=1000;i++) base[i]=base[i-1]*seed;
int T;scanf("%d",&T);
while (T--){
scanf("%d%d",&n,&m);
for (int i=0;i<n;i++)
scanf("%s",s1[i]);
scanf("%d%d",&x,&y);
for (int i=0;i<x;i++)
scanf("%s",s2[i]);
hashs();
//for (int j=0;j<n-x+1;j++){
// for (int i=0;i<m;i++) cout<<hx[j][i]<<" ";
// cout<<endl;
//}
//cout<<endl;
//for (int i=0;i<y;i++) cout<<zx[i]<<" ";
//cout<<endl;
int ans=0;
for (int i=0;i<n-x+1;i++){
ans+=kmp(hx[i],zx,f);
}
printf("%d\n",ans);
}
return 0;
}