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;
}