AC自动机模板 hdu2222

最新推荐文章于 2024-04-26 20:01:35 发布

shine碎

最新推荐文章于 2024-04-26 20:01:35 发布

阅读量162

点赞数

本文链接：https://blog.csdn.net/qq_41006240/article/details/86596557

版权

题目链接

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<new>
#include<queue>
#include<string.h>
using namespace std;
const int sz=250000;
struct node
{
   node *fail;
   node *Next[26];
   int  cnt;
};
node *root;
queue<node *>q;
int ans;
void update(char *s)
{
    int len=strlen(s);
    node *now;
    now=root;
    int i,go,j;
    for(i=0;i<len;i++)
    {
        go=s[i]-'a';
        if(now->Next[go]==NULL)
            {
                now->Next[go]=new node();
                for(j=0;j<26;j++)
                    now->Next[go]->Next[j]=NULL;
                now->Next[go]->fail=NULL;
                now->Next[go]->cnt=0;
            }
        now=now->Next[go];
    }
    now->cnt++;
}
void make_fail()
{
    root->fail=NULL;
    node *now,*p;
    q.push(root);
    while(!q.empty())
    {
        now=q.front();
        q.pop();
        for(int i=0;i<26;i++)
        {
            if(now->Next[i]!=NULL)
            {
              if(now==root)
                    now->Next[i]->fail=root;
              else
              {
                  p=now->fail;
                  while(p!=NULL)
                  {
                      if(p->Next[i]!=NULL)
                        {
                            now->Next[i]->fail=p->Next[i];
                            break;
                        }
                        p=p->fail;
                  }
                  if(p==NULL)
                    now->Next[i]->fail=root;
              }
              q.push(now->Next[i]);
            }
        }

    }
}
void query(char s[])
{
    int len=strlen(s);
    int i,go;
    node *now,*p;
    now=root;
    for(i=0;i<len;i++)
    {
        go=s[i]-'a';
        while(now!=root&&now->Next[go]==NULL)now=now->fail;
        now=now->Next[go];
        if(now==NULL)
            now=root;
        p=now;
        while(p!=root&&p->cnt!=-1)
        {
            ans+=p->cnt;
            p->cnt=-1;
            p=p->fail;
        }
    }
}
void init()
{
    root=new node();
    for(int i=0;i<26;i++)
        root->Next[i]=NULL;
    root->fail=NULL;
    root->cnt=0;
    ans=0;
}
char s[1000007];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        int n,i;

        scanf("%d",&n);
        for(i=0;i<n;i++)
        {
            scanf("%s",s);
            update(s);
        }
        make_fail();
        scanf("%s",s);
        query(s);
        printf("%d\n",ans);
    }
    return 0;
}