AC自动机模板

最新推荐文章于 2014-11-20 15:19:12 发布

fcbruce

最新推荐文章于 2014-11-20 15:19:12 发布

阅读量662

点赞数

分类专栏：数据结构字符串文章标签： ACM algorithm 数据结构 AC自动机

本文链接：https://blog.csdn.net/u012965890/article/details/39781287

版权

数据结构同时被 2 个专栏收录

44 篇文章 0 订阅

订阅专栏

字符串

14 篇文章 0 订阅

订阅专栏

AC自动机，继承自Trie

用于多个模式串的匹配

struct ACauto :public Trie
{
  int cnt;
  int last[maxn];
  int nex[maxn];

  ACauto()
  {
    cnt=0;
    sz=1;
    val[0]=0;
    memset(ch[0],0,sizeof ch[0]);
  }
  void clear()
  {
    cnt=0;
    Trie::clear();
  }

  void calc(int j)
  {
    if (j!=0)
    {
      cnt++;
      calc(last[j]);
    }
  }

  void get_fail()
  {
    int f=0,r=-1;
    nex[0]=0;
    for (int c=0;c<maxsize;c++)
    {
      int u=ch[0][c];
      if (u!=0)
      {
        nex[u]=0;
        q[++r]=u;
        last[u]=0;
      }
    }

    while (f<=r)
    {
      int x=q[f++];
      for (int c=0;c<maxsize;c++)
      {
        int u=ch[x][c];
        if (u==0) continue;
        q[++r]=u;
        int v=nex[x];
        while (v>0 && ch[v][c]==0) v=nex[v];
        nex[u]=ch[v][c];
        last[u]=val[nex[u]]>0?nex[u]:last[nex[u]];
      }
    }
  }

  void find(const char *T)
  {
    for (int i=0,j=0;T[i]!='\0';i++)
    {
      int c=idx(T[i]);
      while (j>0 && ch[j][c]==0) j=nex[j];
      j=ch[j][c];
      if (val[j]!=0)
        calc(j);
      else if (last[j]!=0)
        calc(last[j]);
    }
  }
};

——————————————我是分割线—————————————————

update@2014_11_14

去掉了失配函数，将所有转移一视同仁，方便了在AC自动机上进行DP

struct ACauto :public Trie
{
  int cnt;
  int last[maxn];
  int nex[maxn];

  ACauto()
  {
    cnt=0;
    sz=1;
    val[0]=0;
    memset(ch[0],0,sizeof ch[0]);
  }
  void clear()
  {
    cnt=0;
    Trie::clear();
  }

  void calc(int j)
  {
    if (j!=0)
    {
      cnt++;
      calc(last[j]);
    }
  }

  void get_fail()
  {
    int f=0,r=-1;
    nex[0]=0;
    for (int c=0;c<maxsize;c++)
    {
      int u=ch[0][c];
      if (u!=0)
      {
        nex[u]=0;
        q[++r]=u;
        last[u]=0;
      }
    }

    while (f<=r)
    {
      int x=q[f++];
      for (int c=0;c<maxsize;c++)
      {
        int u=ch[x][c];
        if (u==0)
        {
          ch[x][c]=ch[nex[x]][c];
          continue;
        }
        q[++r]=u;
        int v=nex[x];
        nex[u]=ch[v][c];
        last[u]=val[nex[u]]>0?nex[u]:last[nex[u]];
      }
    }
  }

  void find(const char *T)
  {
    get_fail();
    for (int i=0,j=0;T[i]!='\0';i++)
    {
      int c=idx(T[i]);
      j=ch[j][c];
      if (val[j]!=0)
        calc(j);
      else if (last[j]!=0)
        calc(last[j]);
    }
  }
};