[模板]AC自动机 洛谷3808 AC自动机

题目

分析

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<algorithm>

using namespace std;

struct Tree{
int fail;
int vis[26];
int num;
}ac[1000000];
int n;
int cnt=0;

void Build(string s)
{
int l=s.length();
int now=0;
for(int i=0;i<l;++i)
{
if(ac[now].vis[s[i]-'a']==0) ac[now].vis[s[i]-'a']=++cnt;
now=ac[now].vis[s[i]-'a'];
}
ac[now].num+=1;
}

void Get_fail()
{
queue<int> Q;
for(int i=0;i<26;++i)
{
if(ac[0].vis[i]!=0)
{
ac[ac[0].vis[i]].fail=0;
Q.push(ac[0].vis[i]);
}
}
while(!Q.empty())
{
int u=Q.front();
Q.pop();
for(int i=0;i<26;++i)
{
if(ac[u].vis[i]!=0)
{
ac[ac[u].vis[i]].fail=ac[ac[u].fail].vis[i];
Q.push(ac[u].vis[i]);
}
else ac[u].vis[i]=ac[ac[u].fail].vis[i];
}
}
}

int ac_Query(string s)
{
int l=s.length();
int now=0,ans=0;
for(int i=0;i<l;++i)
{
now=ac[now].vis[s[i]-'a'];
for(int t=now;t&&(ac[t].num!=-1);t=ac[t].fail)
{
ans+=ac[t].num;
ac[t].num=-1;
}
}
return ans;
}

int main()
{
cin>>n;
for(int i=1;i<=n;++i)
{
string s;
cin>>s;
Build(s);
}
ac[0].fail=0;
Get_fail();
string s;
cin>>s;
cout<<ac_Query(s)<<endl;
return 0;
}