洛谷 P3808 【模板】AC自动机（简单版）洛谷 P3796 【模板】AC自动机（加强版）

https://www.cnblogs.com/gtarcoder/p/4820560.html

每个节点的后缀指针fail指针指向：

例如he,she,his,hers的例子（见蓝书P214）：

7号点表示串his，将其头部去掉得到is但不存在该节点，再次将其头部去掉得到s，存在该节点，因此7号的后缀指针指向表示s的3号

5号点表示串she，将其头部去掉得到he，存在该节点，因此5号的后缀指针指向表示he的2号

从根节点到节点P可以得到一个字符串S，节点P的前缀指针定义为 指向树中出现过的S的最长后缀（不能等于S）

可以将trie上所有不存在的边u--(ch)-->v都补全为其fail指针指向的同类型边（f[u]--(ch)-->v'得到u--(ch)-->v'）。（减少特判且方便递推）

每个节点的lst，与fail的区别在于，其指向的点表示的字符串一定是原来字符串集合中某一个完整的串（或空串），而fail则只要求原集合中某串的前缀即可

P3808中：由于只计算“出现过的”，所以某个算过贡献后要赋为0

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<queue>
 using namespace std;
 char *s[],tttt[],*sp=tttt,s2[];
 int n;
 //namespace AC
 //{
 //    int ch[1001000][26],sum[1001000],f[1001000],root,mem,lst[1000100];
 //    void insert(char *x)
 //    {
 //        if(!root)    root=++mem;
 //        int now=root,u;
 //        for(;*x;x++)
 //        {
 //            u=*x-'a';
 //            if(!ch[now][u])    ch[now][u]=++mem;
 //            now=ch[now][u];
 //        }
 //        sum[now]++;
 //    }
 //    void build()
 //    {
 //        int c,t,u,v;
 //        queue<int> q;
 //        f[root]=root;
 //        for(c=0;c<26;c++)
 //        {
 //            u=ch[root][c];
 //            if(u)    {f[u]=root;q.push(u);lst[u]=root;}
 //        }
 //        while(!q.empty())
 //        {
 //            t=q.front();q.pop();
 //            for(c=0;c<26;c++)
 //            {
 //                u=ch[t][c];
 //                if(!u)    {ch[t][c]=ch[f[t]][c];continue;}
 //                q.push(u);
 //                v=f[t];
 //                while(v!=root&&!ch[v][c])    v=f[v];
 //                f[u]=ch[v][c];
 //                lst[u]=sum[u]?f[u]:lst[f[u]];
 //            }
 //        }
 //    }
 //    void search(char *x)
 //    {
 //        int j=root;
 //        for(;*x;x++)
 //        {
 //            c=*x-'a';
 //            if(sum[c])
 //        }
 //    }
 //}
 //注释掉的是root任意值的情况，未完成，以下程序认为root为0号点
 namespace AC
 {
     int ch[][],sum[],f[],mem,lst[];
     void insert(char *x)
     {
         int now=,u;
         for(;*x;x++)
         {
             u=*x-'a';
             if(!ch[now][u])    ch[now][u]=++mem;
             now=ch[now][u];
         }
         sum[now]++;
     }
     void build()
     {
         int c,t,u;
         queue<int> q;
         f[]=;
         for(c=;c<;c++)
         {
             u=ch[][c];
             if(u)    {f[u]=;q.push(u);lst[u]=;}
         }
         while(!q.empty())
         {
             t=q.front();q.pop();
             for(c=;c<;c++)
             {
                 u=ch[t][c];
                 if(!u)    {ch[t][c]=ch[f[t]][c];continue;}
                 q.push(u);
                 f[u]=ch[f[t]][c];
                 lst[u]=sum[f[u]]?f[u]:lst[f[u]];
             }
         }
     }
     int ans;
     void get_ans(int j)
     {
         while(j)
         {
             ans+=sum[j];
             sum[j]=;
             j=lst[j];
         }
     }
     int count(char *x)
     {
         ans=;
         int j=,c;
         for(;*x;x++)
         {
             c=*x-'a';
             j=ch[j][c];
             get_ans(j);
         }
         return ans;
     }
 }
 int main()
 {
     int i;
     scanf("%d",&n);
     for(i=;i<=n;i++)
     {
         s[i]=sp;scanf("%s",s[i]);sp+=strlen(s[i])+;
         AC::insert(s[i]);
     }
     AC::build();
     scanf("%s",s2);
     printf("%d",AC::count(s2));
     return ;
 }

---

P3796

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<queue>
 #include<vector>
 #include<map>
 #include<string>
 using namespace std;
 char *s[],tttt[],*sp,s2[];
 int n;
 namespace AC
 {
     int ch[][],sum[],f[],mem,lst[],ans[];
     void clear()
     {
         int i,j;
         for(i=;i<=mem;i++)
         {
             for(j=;j<;j++)    ch[i][j]=;
             sum[i]=f[i]=lst[i]=ans[i]=;
         }
         mem=;
     }
     int insert(char *x)
     {
         int now=,u;
         for(;*x;x++)
         {
             u=*x-'a';
             if(!ch[now][u])    ch[now][u]=++mem;
             now=ch[now][u];
         }
         sum[now]++;
         return now;
     }
     void build()
     {
         int c,t,u;
         queue<int> q;
         f[]=;
         for(c=;c<;c++)
         {
             u=ch[][c];
             if(u)    {f[u]=;q.push(u);lst[u]=;}
         }
         while(!q.empty())
         {
             t=q.front();q.pop();
             for(c=;c<;c++)
             {
                 u=ch[t][c];
                 if(!u)    {ch[t][c]=ch[f[t]][c];continue;}
                 q.push(u);
                 f[u]=ch[f[t]][c];
                 lst[u]=sum[f[u]]?f[u]:lst[f[u]];
             }
         }
     }
     void get_ans(int j)
     {
         while(j)
         {
             ans[j]+=sum[j];
             j=lst[j];
         }
     }
     void work(char *x)
     {
         int j=,c;
         for(;*x;x++)
         {
             c=*x-'a';
             j=ch[j][c];
             get_ans(j);
         }
     }
 }
 int pos[];
 int a1,anss;
 int main()
 {
     int i;
     while()
     {
         scanf("%d",&n);
         if(n==)    break;
         AC::clear();
         sp=tttt;
         for(i=;i<=n;i++)
         {
             s[i]=sp;scanf("%s",s[i]);sp+=strlen(s[i])+;
             pos[i]=AC::insert(s[i]);
         }
         AC::build();
         scanf("%s",s2);
         AC::work(s2);a1=anss=;
         for(i=;i<=n;i++)
             if(AC::ans[pos[i]]>a1)
                 a1=AC::ans[pos[i]];
         for(i=;i<=n;i++)
             if(AC::ans[pos[i]]==a1)
                 anss++;
         printf("%d\n",a1);
         for(i=;i<=n;i++)
             if(AC::ans[pos[i]]==a1)
                 printf("%s\n",s[i]);
     }
     return ;
 }