UVA-11019 Matrix Matcher 【AC自动机二维匹配】

题意：给两个由字符串构成的矩阵A,B  求B在A中出现次数。

 

如果是一维的话很简单，现在考虑如何通过一维的状态转移到二维。

B中每行的位置是确定的，那么将B矩阵的每行的串全部压入tire树中，对A的每一行进行匹配，匹配成功时就将信息存到B的右上角，最后若是B的右上角满足，res+1。tire树中开一个数组储存一下行数即可。

具体见注释

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define LOGN 10
#define MAXN (1<<LOGN)
#define MAXNODES 3*( (1<<(2*LOGN)) / 4 + 100)
#define son(x) (p*4-2+x)
using namespace std;
const int mod = 20071027;
struct node
{
    node* fail;
    node* las;
    node* next[26];
    int c[104];//储存这个串在B中的第几行，因为可能有多行重复，所以开数组
    int cnt;
};
int ans[1004][1004];//储存整个矩阵中，每个有可能被匹配的小矩阵的右上角的信息
node* rt;
inline void init(node* rt)
{
    rt->cnt=0;
    rt->fail=NULL;
    rt->las=NULL;
    for (int i = 0; i < 26; ++i) {
        rt->next[i]=NULL;
    }
    memset(rt->c,0, sizeof(rt->c));
}
void insert(char *s,int id)//id表示第几行
{
    int len=strlen(s);
    node* now = rt;
    for (int i = 0; i < len; ++i) {
        int v = s[i] - 'a';
        if(now->next[v]==NULL)
        {
            node* t = new node;
            init(t);
            now->next[v]=t;
        }
        now=now->next[v];
    }
    now->c[now->cnt]=id;//存入数组
    now->cnt++;
}
void acauto(node* rt)
{
    queue<node*>q;
    q.push(rt);
    while(!q.empty())
    {
        node* t=q.front();q.pop();
        for (int i = 0; i < 26; ++i) {
            if(t->next[i]!=NULL)
            {
                if(t==rt){
                    t->next[i]->fail=rt;
                    t->next[i]->las=rt;
                }
                else
                {
                    node *p=t->fail;
                    while(p!=NULL)
                    {
                        if(p->next[i]!=NULL)
                        {
                            t->next[i]->fail=p->next[i];
                            break;
                        }
                        p=p->fail;
                    }
                    if(p==NULL)t->next[i]->fail=rt;
                    else{
                        if(p->next[i]->cnt>0) t->next[i]->las=p->next[i];
                        else t->next[i]->las=p->next[i]->las;
                    }
                }
                q.push(t->next[i]);
            }
        }
    }
}
int res;
int n,m;
int x,y;
void querry(node* rt,char* s,int id)
{

    int len=strlen(s);
    node* p =rt;
    for (int i = 0; i < len; ++i) {//此时的i可以表示列
        int v=s[i]-'a';
        while(p->next[v]==NULL&&p!=rt) p=p->fail;
        p=(p->next[v]==NULL?rt:p->next[v]);
        node* t=p;
        while(t!=NULL&&t!=rt)
        {
            if(t->cnt>=0)
            {
                for (int j = 0; j < t->cnt; ++j) {//匹配成功，遍历数组，将所有的信息更新到对应的点上
                    if(id-t->c[j]<0)continue;
                    ans[i][id-t->c[j]]++;//右上角的坐标可以算出来
                    if(ans[i][id-t->c[j]]==x)res++;//右上角加满x行，res++
                }
                //t->cnt=-1;
            }
            else break;
            t=t->las;
        }
    }
}
void free(node *t)
{
    if(t==NULL)return;;
    for (int i = 0; i < 26; ++i) {
        free(t->next[i]);
    }
    delete(t);
}
char s[1004][1004];
char ts[104];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        rt=new node;
        init(rt);

        scanf("%d%d",&n,&m);
        for (int i = 0; i < n; ++i) {
            scanf("%s",s[i]);
        }
        scanf("%d%d",&x,&y);
        for (int i = 0; i < x; ++i) {
            scanf("%s",ts);
            insert(ts,i);
        }
        acauto(rt);
        res=0;memset(ans,0, sizeof(ans));//没清空wa了一次
        for (int i = 0; i < n; ++i) {
            querry(rt,s[i],i);
        }
        printf("%d\n",res);
        free(rt);
    }


}

/*
uva 3942
poweroj 2806
 */
/*
3
3 3
abc
bcd
cde
2 2
bc
cd
4 4
abba
baab
abab
baba
2 2
ab
ab
1 1
x
1 1
y
3 3
abc
bcd
cde
2 2
bc
cd
*/