【uva11019-Matrix Matcher】AC自动机+优化+记录

http://acm.hust.edu.cn/vjudge/problem/33057

题意：在二维文本串T中查找一个二维模板串P出现了多少次。

题解：

拆分模板串P的每一行，建AC自动机。
拆分文本串T的每一行，在自动机中与P匹配，ct[i][j]表示以点(i,j)为左上角、与P等大的矩形有多少个对应的行与P匹配。
最后ct[i][j]==P的行数的i，j就是一个匹配点，ans++。
注意：1.原本我在trie的叶子用动态数组维护了一个表示这一行是P的第几行的数组，但是超时了，后来看了LRJ的代码，改成了用一个nt[i]来表示重复的行的下一行，就A了。
2.注意在ct[i][j]里加的时候判断i,j是否大于0.

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<queue>
 #include<vector>
 using namespace std;

 int n,m,num,ans;
 int ct[][],nt[];
 char s[][],ss[][];
 struct node{
     int son[];
     int fail,ed;
 }a[*];
 queue<int> q;

 void clear(int x)
 {
     a[x].fail=a[x].ed=;
     memset(a[x].son,,sizeof(a[x].son));
 }

 void trie(char *c,int id)
 {
     int l=strlen(c);
     int x=;
     for(int i=;i<l;i++)
     {
         int t=c[i]-'a'+;
         if(!a[x].son[t])
         {
             num++;
             clear(num);
             a[x].son[t]=num;
         }
         x=a[x].son[t];
     }
     if(!a[x].ed) a[x].ed=id;
 }

 void buildAC()
 {
     while(!q.empty()) q.pop();
     for(int i=;i<=;i++)
         if(a[].son[i]) q.push(a[].son[i]);
     while(!q.empty())
     {
         int x=q.front();q.pop();
         int fail=a[x].fail;
         for(int i=;i<=;i++)
         {
             int y=a[x].son[i];
             if(y)
             {
                 a[y].fail=a[fail].son[i];
                 q.push(y);
             }
             else a[x].son[i]=a[fail].son[i];
         }
     }
 }

 void find(char *c,int id)
 {
     int l=strlen(c);
     int x=;
     for(int i=;i<l;i++)
     {
         int t=c[i]-'a'+;
         if(!a[x].son[t]) x=;
         x=a[x].son[t];
         int x0=id,y0=(i+)-m+;
         int p=a[x].ed;
         while(p)
         {
             int xx=x0-p+,yy=y0;
             if(xx>= && yy>=)//注意！！没有就WA！！
             {
                 ct[xx][yy]++;
                 if(ct[xx][yy]>=n) ans++;
             }
             p=nt[p];
         }
     }
 }

 int main()
 {
     freopen("a.in","r",stdin);
     freopen("a.out","w",stdout);
     int T;
     scanf("%d",&T);
     while(T--)
     {
         num=ans=;
         clear();
         memset(ct,,sizeof(ct));
         memset(nt,,sizeof(nt));//用nt数组（next）优化，不然就TLE。
         int x,y;
         scanf("%d%d",&x,&y);
         for(int i=;i<=x;i++)
         {
             scanf("%s",ss[i]);
         }
         scanf("%d%d",&n,&m);
         for(int i=;i<=n;i++)
         {
             scanf("%s",s[i]);
             trie(s[i],i);
         }
         for(int i=;i<=n;i++)
             for(int j=i+;j<=n;j++)
             {
                 if(strcmp(s[i],s[j])==) {nt[i]=j;break;}
             }
         buildAC();
         for(int i=;i<=x;i++) find(ss[i],i);
         printf("%d\n",ans);
     }
     return ;
 }