Matrix Matcher（AC自动机/哈希、二维匹配）

题目链接：https://vjudge.net/problem/UVA-11019

给定一个N×M矩阵，你的任务是找出一个X×Y模式的出现次数。

思路：要匹配模式矩阵，我们可以将模式矩阵中的X个串放入tire树中，同时记录下这个串所在的行数。然后我们开始判断以N×M矩阵中每个点为左上角的X*Y大小的矩阵是否匹配，每匹配到一个模式串X[i]，就在它所对应的这个矩形的左上角位置上+1。

如上图匹配到X[1]和X[3]时，都应该在(1,1)位置上+1。这样匹配完后，所有值为X的点就代表一个匹配成功的子矩阵了。

//最后重要的是发现这个，把待判信息都存左上角，处理完后再判断“答案”是否正确，的思路和前几天看的一道题思路似乎一样：https://codeforces.com/contest/1294/problem/E

代码：

#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
//#define int long long
const int manx=1e4+1000;
const int max_len=2e6+10;
char s[manx/10][manx/10],p[manx/100][manx/100];
int vss[manx/10][manx/10],vt[manx/100][manx/100],n,m;
struct AC
{
    int tree[manx][26],fail[manx],vs[manx];
    int cou;
    int newnode()
    {
        for(int i=0;i<26;i++)
            tree[cou][i]=0;
        vs[cou]=fail[cou++]=0;
        return cou-1;
    }
    void init()
    {
        cou=0;
        newnode();
        memset(vss,0,sizeof(vss));
    }
    void add(char *s,int line)
    {
        int u=0;
        for(int i=0;s[i]!='\0';i++)
        {
            int net=s[i]-'a';
            if(!tree[u][net])tree[u][net]=newnode();
            u=tree[u][net];
        }
        vs[u]++;//可能对应模式矩阵中的几个子串
        vt[u][vs[u]]=line;
    }
    void getfail()
    {
        int u=0;
        queue<int>qu;
        for(int i=0;i<26;i++)
            if(tree[u][i])
                qu.push(tree[u][i]);
        while(!qu.empty())
        {
            u=qu.front();
            qu.pop();
            for(int i=0;i<26;i++)
            {
                if(tree[u][i])
                {
                    fail[tree[u][i]]=tree[fail[u]][i];
                    qu.push(tree[u][i]);
                }
                else
                    tree[u][i]=tree[fail[u]][i];
            }
        }
    }
    void query(char* s,int line)
    {
        int u=0;
        for(int i=0;s[i]!='\0';i++)
        {
            int net=s[i]-'a';
            u=tree[u][net];
            for(int j=1;j<=vs[u];j++)//可能对应模式矩阵中的几个子串
            {
                int x=line-vt[u][j]+1;
                int y=i-m+1;//求它所对应的左上角坐标
                if(x>=0&&y>=0)
                    vss[x][y]++;
            }
        }
    }
};
AC aut;
int main()
{
    int t,x,y;
    scanf("%d",&t);
    while(t--)
    {
        aut.init();
        scanf("%d%d",&x,&y);
        for(int i=0;i<x;i++)
            scanf("%s",s[i]);
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            scanf("%s",p[i]);
            aut.add(p[i],i);
        }
        aut.getfail();
        for(int i=0;i<x;i++)
            aut.query(s[i],i);
        int ans=0;
        for(int i=0;i<x;i++)
            for(int j=0;j<y;j++)
                if(vss[i][j]==n)
                    ans++;
        printf("%d\n",ans);
    }
    return 0;
}

另外还有哈希的做法：

我们可以先处理出N×M矩阵中每个位置到它前面的Y个字符的哈希值，以及模式矩阵每一行的哈希值。后面匹配时匹配到X[0]，就判断下面的X-1行是否匹配。

代码：

#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
#define LL long long
const int manx=1e3+10;
const int max_len=2e6+10;
const LL mod=1e9+7;
const LL mmm=1e7+19;
char s[manx][manx],p[manx/10][manx/10];
LL sval[manx][manx],pval[manx/10],mul[manx];
int main()
{
    int t,x,y,n,m;
    scanf("%d",&t);
    mul[0]=1;
    for(int i=1;i<=1000;i++)
        mul[i]=(mul[i-1]*mmm)%mod;
    while(t--)
    {
        scanf("%d%d",&x,&y);
        for(int i=0;i<x;i++)
            scanf("%s",s[i]);
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            scanf("%s",p[i]);
            pval[i]=p[i][0];
            for(int j=1;j<m;j++)
                pval[i]=(pval[i]*mmm+p[i][j])%mod;
        }
        for(int i=0;i<x;i++)
        {
            sval[i][0]=s[i][0];
            for(int j=1;j<y;j++)
            {
                if(j<m)
                   sval[i][j]=(sval[i][j-1]*mmm+s[i][j])%mod;
                else
                   sval[i][j]=(mmm*(sval[i][j-1]-s[i][j-m]*mul[m-1]%mod+mod)%mod+s[i][j])%mod;
            }
        }
        int ans=0;
        for(int i=0;i<=x-n;i++)
        {
            for(int j=m-1;j<y;j++)
            {
                if(sval[i][j]==pval[0])
                {
                    int cou=1;
                    while(cou<n&&sval[i+cou][j]==pval[cou])cou++;
                    if(cou==n)ans++;
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}