#include<bits/stdc++.h>
using namespace std;
#define ll long long
int ans=0;
const int maxnode=1e6+7;
const int sigma_size=26;
struct AC_Automata
{
int ch[maxnode][sigma_size];
int val[maxnode]; // 每个字符串的结尾结点都有一个非0的val
int f[maxnode]; // fail函数
int last[maxnode]; // last[i]=j表j节点表示的单词是i节点单词的后缀,且j节点是单词节点
int sz;
//初始化0号根节点的相关信息
void init()
{
sz=1;
memset(ch[0],0,sizeof(ch[0]));
val[0]=0;
}
void insert(char *s,int v)
{
int u=0;
for(int i=0; s[i]; i++)
{
int id=s[i]-'a';
if(ch[u][id]==0)
{
ch[u][id]=sz;
memset(ch[sz],0,sizeof(ch[sz]));
val[sz++]=0;
}
u=ch[u][id];
}
val[u]+=v;
}
void getFail()
{
queue<int> q;
last[0]=f[0]=0;
for(int i=0; i<sigma_size; i++)
{
int u=ch[0][i];
if(u)
{
f[u]=last[u]=0;
q.push(u);
}
}
while(!q.empty())// 按BFS顺序计算fail
{
int r=q.front(); q.pop();
for(int i=0; i<sigma_size; i++)
{
int u=ch[r][i];
if(u==0)continue;
q.push(u);
int v=f[r]; //失配边尝试平行传递
while(v && ch[v][i]==0) v=f[v];
f[u]= ch[v][i];
last[u] = val[f[u]]?f[u]:last[f[u]];
}
}
}
//递归打印与结点i后缀相同的前缀节点编号
//进入此函数前需保证val[i]>0
void print(int i)
{
if(i)
{
ans+=val[i];
val[i]=0;
print(last[i]);
}
}
void find(char *s)
{
int j=0;//j:trie树中的节点编号
for(int i=0; s[i]; i++)
{
int id=s[i]-'a';
while(j && ch[j][id]==0) j=f[j];//ch[j][id]:第j个节点的id字母子节点编号
j=ch[j][id];
if(val[j]) print(j);
else if(last[j]) print(last[j]);
}
}
};
AC_Automata ac;
const int maxn=1e6+7;
char ori[maxn];
char tar[maxn];
int main(){
int t;
scanf("%d",&t);
while(t--){
ans=0;
ac.init();
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%s",tar);
ac.insert(tar,1);
}
ac.getFail();
scanf("%s",ori);
ac.find(ori);
printf("%d\n",ans);
}
}
自动机的英文念起来好中二啊,
自动机就是trie树和kmp算法的组合,多模板匹配的首选算法
两种快乐的算法重叠在一起,就有了自动机(
-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
贴一个轻量化的板子,原题HDU3065
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=2e6+7;
const int maxm=55;
const int maxnode=1010*51;
const int sigma=128-30;
int n,m;
int vis[1010];
char word[1010][55];
char ori[maxn];
struct automata{
int ch[maxnode][sigma];
int val[maxnode];
int f[maxnode];
int sz;
int newnode(){
memset(ch[sz],0,sizeof(ch[sz]));
f[sz]=val[sz]=0;
return sz++;
}
void init(){
memset(val,0,sizeof(val));
sz=0;
newnode();
}
void insert(char s[],int v){
int u=0;
for(int i=0;s[i];i++){
int &x=ch[u][s[i]-30];
u=x?x:x=newnode();
}
val[u]=v;
}
void build(){
queue<int>q;
q.push(0);
while(!q.empty()){
int u=q.front();q.pop();
for(int i=0;i<sigma;i++){
int v=ch[u][i];
if(!v)ch[u][i]=ch[f[u]][i];
else q.push(v);
if(u&&v)f[v]=ch[f[u]][i];
}
}
}
int find(char s[]){
int j=0;
for(int i=0;s[i];i++){
int id=s[i]-30;
j=ch[j][id];
int tmp=j;
while(tmp){
vis[val[tmp]]++;
tmp=f[tmp];
}
}
}
}ac;
int main(){
while(~scanf("%d",&n)){
memset(vis,0,sizeof(vis));
ac.init();
for(int i=1;i<=n;i++){
scanf("%s",word[i]);
ac.insert(word[i],i);
}
ac.build();
scanf("%s",ori);
ac.find(ori);
for(int i=1;i<=n;i++){
if(vis[i])printf("%s: %d\n",word[i],vis[i]);
}
}
}