【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.

  1 #include<cstdio>
  2 #include<cstdlib>
  3 #include<cstring>
  4 #include<iostream>
  5 #include<queue>
  6 #include<vector>
  7 using namespace std;
  8
  9 int n,m,num,ans;
 10 int ct[1100][1100],nt[110];
 11 char s[1100][1100],ss[1100][1100];
 12 struct node{
 13     int son[30];
 14     int fail,ed;
 15 }a[110*110];
 16 queue<int> q;
 17
 18 void clear(int x)
 19 {
 20     a[x].fail=a[x].ed=0;
 21     memset(a[x].son,0,sizeof(a[x].son));
 22 }
 23
 24 void trie(char *c,int id)
 25 {
 26     int l=strlen(c);
 27     int x=0;
 28     for(int i=0;i<l;i++)
 29     {
 30         int t=c[i]-‘a‘+1;
 31         if(!a[x].son[t])
 32         {
 33             num++;
 34             clear(num);
 35             a[x].son[t]=num;
 36         }
 37         x=a[x].son[t];
 38     }
 39     if(!a[x].ed) a[x].ed=id;
 40 }
 41
 42 void buildAC()
 43 {
 44     while(!q.empty()) q.pop();
 45     for(int i=1;i<=26;i++)
 46         if(a[0].son[i]) q.push(a[0].son[i]);
 47     while(!q.empty())
 48     {
 49         int x=q.front();q.pop();
 50         int fail=a[x].fail;
 51         for(int i=1;i<=26;i++)
 52         {
 53             int y=a[x].son[i];
 54             if(y)
 55             {
 56                 a[y].fail=a[fail].son[i];
 57                 q.push(y);
 58             }
 59             else a[x].son[i]=a[fail].son[i];
 60         }
 61     }
 62 }
 63
 64 void find(char *c,int id)
 65 {
 66     int l=strlen(c);
 67     int x=0;
 68     for(int i=0;i<l;i++)
 69     {
 70         int t=c[i]-‘a‘+1;
 71         if(!a[x].son[t]) x=0;
 72         x=a[x].son[t];
 73         int x0=id,y0=(i+1)-m+1;
 74         int p=a[x].ed;
 75         while(p)
 76         {
 77             int xx=x0-p+1,yy=y0;
 78             if(xx>=1 && yy>=1)//注意！！没有就WA！！
 79             {
 80                 ct[xx][yy]++;
 81                 if(ct[xx][yy]>=n) ans++;
 82             }
 83             p=nt[p];
 84         }
 85     }
 86 }
 87
 88 int main()
 89 {
 90     freopen("a.in","r",stdin);
 91     freopen("a.out","w",stdout);
 92     int T;
 93     scanf("%d",&T);
 94     while(T--)
 95     {
 96         num=ans=0;
 97         clear(0);
 98         memset(ct,0,sizeof(ct));
 99         memset(nt,0,sizeof(nt));//用nt数组（next）优化，不然就TLE。
100         int x,y;
101         scanf("%d%d",&x,&y);
102         for(int i=1;i<=x;i++)
103         {
104             scanf("%s",ss[i]);
105         }
106         scanf("%d%d",&n,&m);
107         for(int i=1;i<=n;i++)
108         {
109             scanf("%s",s[i]);
110             trie(s[i],i);
111         }
112         for(int i=1;i<=n;i++)
113             for(int j=i+1;j<=n;j++)
114             {
115                 if(strcmp(s[i],s[j])==0) {nt[i]=j;break;}
116             }
117         buildAC();
118         for(int i=1;i<=x;i++) find(ss[i],i);
119         printf("%d\n",ans);
120     }
121     return 0;
122 }