AC自动机

Posted 2020-09-26 TS_Hugh

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define L 51
#define LL 100000002
using namespace std;
char c[11][L],s[LL];
struct Trie
{
    Trie *ch[26],*fail,*last;
    int who;
    Trie()
    {
       memset(ch,0,sizeof(ch));
       fail=NULL;
       last=NULL;
       who=0;
    }
    void* operator new(size_t size);
}*root,*C,*mempool,*q[L*11];
void* Trie :: operator new(size_t size)
{
    if(C==mempool)
    {
       C=new Trie[(1<<15)+10];
       mempool=C+(1<<15)+10;
    }
    return C++;
}
int n;
inline void insert(char *S,int x)
{
    int len=strlen(S);
    Trie *now=root;
    for(int i=0;i<len;i++)
    {
      if(now->ch[S[i]-‘a‘]==NULL)
       now->ch[S[i]-‘a‘]=new Trie;
      now=now->ch[S[i]-‘a‘];
    }
    now->who=x;
}
inline void build()
{
    q[0]=root;
    for(int i=0,j=0;i<=j;i++)
     for(int l=0;l<26;l++)
      if(q[i]->ch[l])
      {
         Trie *now=q[i]->fail;
         while(now&&!now->ch[l])now=now->fail;
         q[i]->ch[l]->fail=now?now->ch[l]:root;
         q[++j]=q[i]->ch[l];
         q[j]->last=q[j]->fail->who?q[j]->fail:q[j]->fail->last;
      }
}
inline void Init()
{
  root=new Trie;
  scanf("%d",&n);
  for(int i=1;i<=n;i++)   
  {
    scanf("%s",c[i]);
    insert(c[i],i);
  }  
  scanf("%s",s);
  build();
}
int sum[11];
inline void work()
{
  Trie *now=root;
  for(int i=0;s[i];i++)
  {
    int to=s[i]-‘a‘;
    while(now&&!now->ch[to])now=now->fail;
    now=now?now->ch[to]:root;
    for(Trie *Now=now;Now;Now=Now->last)
     sum[Now->who]++;
  }
}
inline void print()
{
  for(int i=1;i<=n;i++)
    printf("%s %d\n",c[i],sum[i]);
}
int main()
{
  freopen("ACautomata.in","r",stdin);
  freopen("ACautomata.out","w",stdout);
  Init();
  work();
  print();
  return 0;
}