UVa 11019 Matrix Matcher

题目链接

UVa 11019 Matrix Matcher

题解

这种二维化一维的方法并不少见但是我觉得我还是想不出来.

字符矩阵P可以先拆成很多行然后合成一个AC自动机,然后去匹配T的每一行,然后用cnt[i][j]表示以(i,j)为左上角这个大小为P的矩阵里有多少行与P匹配,那如果达到x了的话就是一个匹配点,当在T的第r行匹配到P的第i行(也就是AC自动机里面编为i的模板),最左边一列为c时那么cnt[r-i][c]应该++,(这里要求从0开始编号)注意判断r-i>=0;还有一个很重要的地方是模板可能会相同,那么在build trie的时候就需要搞一点事情,把每个节点串成一条链,因为模板编号是惟一的,所以可以用next[i]表示与i相同的下一个模板的编号,刚开始将所有trie节点的val置为-1.

代码

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
const int maxx=100+10;
const int maxn=1000+10;
const int sigma_size=26;
const int maxnode=100*100+10;
int idx(char ch){
    return ch-'a';
}
int read(){
    int ret=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    for(;ch>='0'&&ch<='9';ch=getchar()) ret=ret*10+ch-'0';
    return ret;
}

int n,m,x,y,next[maxn];
struct AC_auto{
    int sz,val[maxnode];
    int ch[maxnode][sigma_size];
    inline void clear(){
        sz=1;val[0]=-1;
        memset(ch[0],0,sizeof(ch[0]));
        memset(count,0,sizeof(count));
    }
    inline void insert(char *T,int v){
        int n=strlen(T),u=0;
        for(int i=0;i<n;i++)
        {
            int c=idx(T[i]);
            if(!ch[u][c]){
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=-1;ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        next[v]=val[u];val[u]=v;
    }
    int f[maxnode],count[maxn][maxn];
    inline void getfail(){
        queue<int>q;
        f[0]=0;
        for(int i=0;i<sigma_size;i++)
        {
            int x=ch[0][i];
            if(x) f[x]=0,q.push(x);
        }
        while(!q.empty())
        {
            int r=q.front();q.pop();
            for(int c=0;c<sigma_size;c++)
            {
                int u=ch[r][c];
                if(!u) {
                    ch[r][c]=ch[f[r]][c];continue;
                }
                q.push(u);
                int v=f[r];
                while(v&&!ch[v][c]) v=f[v];
                f[u]=ch[v][c];
            }
        }
        return ;
    }
    inline void deal(int cur,int j,int i){
        for(;j!=-1;j=next[j])
          if(cur>=j) count[cur-j][i]++;
    }
    inline void find(int cur,char *T){
        int n=strlen(T),j=0;
        for(int i=0;i<n;i++)
        {
            int c=idx(T[i]);
            j=ch[j][c];
            if(val[j]!=-1) deal(cur,val[j],i-y+1);
        }
    }
}ac;

char T[maxn][maxn],S[maxx][maxx];
inline void solve()
{
    int ans=0;ac.clear();
    n=read(),m=read();
    rep(i,n) scanf("%s",T[i]);
    x=read(),y=read();
    rep(i,x) {
        scanf("%s",S[i]);
        ac.insert(S[i],i);  
    }
    ac.getfail();
    rep(i,n) ac.find(i,T[i]);
    rep(i,n) rep(j,n) if(ac.count[i][j]==x) ans++;
    printf("%d\n",ans);
    return ;
}

int main()
{
    int T;cin>>T;
    while(T--) solve();
    return 0;
}