luogu AC自动机(模板)
完全忘了AC自动机怎么写了qwq,更别说AC自动机上DP了。
今天就好好地学习字符串好了qwq
提一下AC自动机的时间复杂度——设n是模式串的个数,m是文本串的长度,l是模式串的平均长度,那么它的时间复杂度就是\(O(n+m)*l\)。
AC自动机上fail指针指向的点,从root到它代表的是其后缀。
简单版:给定一个文本串,和多个模式串。求有多少个模式串在文本串中出现过。qwqwq
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>
#define MAXN 1000010
using namespace std;
int cnt,n;
char s[MAXN];
struct tree{int fail,end,t[26];}ac[MAXN];
inline void build(char s[])
{
int len=strlen(s);
int now=0;
for(int i=0;i<len;i++)
{
if(!ac[now].t[s[i]-'a'])
ac[now].t[s[i]-'a']=++cnt;
now=ac[now].t[s[i]-'a'];
}
ac[now].end+=1;
}
inline void get_fail()
{
queue<int>q;
for(int i=0;i<=25;i++)
if(ac[0].t[i]!=0)
ac[ac[0].t[i]].fail=0,q.push(ac[0].t[i]);
while(!q.empty())
{
int u=q.front();q.pop();
for(int i=0;i<=25;i++)
{
if(ac[u].t[i]!=0)
{
ac[ac[u].t[i]].fail=ac[ac[u].fail].t[i];
q.push(ac[u].t[i]);
}
else ac[u].t[i]=ac[ac[u].fail].t[i];
}
}
}
inline int ac_query(char s[])
{
int len=strlen(s);
int now=0,ans=0;
for(int i=0;i<len;i++)
{
int now=ac[now].t[s[i]-'a'];
for(int k=now;k&&ac[k].end!=-1;k=ac[k].fail)
ans+=ac[k].end,ac[k].end=-1;
}
return ans;
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("ce.in","r",stdin);
#endif
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%s",s);
build(s);
}
ac[0].fail=0;
get_fail();
scanf("%s",s);
printf("%d\n",ac_query(s));
}
加强版:给定一个文本串和许多模式串,问在文本串中出现次数最多的模式串?以及出现的次数。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#define MAXN 1000010
using namespace std;
int cnt,n;
char s[1000][100],q[MAXN];
struct tree{int fail,end,t[26];}ac[MAXN];
struct Ans{int num,pos;}ans[MAXN];
inline bool cmp(struct Ans x,struct Ans y)
{
if(x.num!=y.num) return x.num>y.num;
else return x.pos<y.pos;
}
inline void clean(int x)
{
memset(ac[x].t,0,sizeof(ac[x].t));
ac[x].fail=ac[x].end=0;
}
inline void build(char s[],int num)
{
int len=strlen(s),now=0;
for(int i=0;i<len;i++)
{
if(ac[now].t[s[i]-'a']==0)
ac[now].t[s[i]-'a']=++cnt,clean(cnt);
now=ac[now].t[s[i]-'a'];
}
ac[now].end=num;
}
inline void get_fail()
{
queue<int>q;
for(int i=0;i<=25;i++)
{
if(ac[0].t[i]!=0)
ac[ac[0].t[i]].fail=0,q.push(ac[0].t[i]);
}
while(!q.empty())
{
int u=q.front();q.pop();
for(int i=0;i<=25;i++)
{
if(ac[u].t[i]!=0)
{
ac[ac[u].t[i]].fail=ac[ac[u].fail].t[i];
q.push(ac[u].t[i]);
}
else ac[u].t[i]=ac[ac[u].fail].t[i];
}
}
}
inline void ac_query(char s[])
{
int len=strlen(s),now=0;
for(int i=0;i<len;i++)
{
now=ac[now].t[s[i]-'a'];
for(int k=now;k;k=ac[k].fail)
ans[ac[k].end].num++;
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("ce.in","r",stdin);
#endif
scanf("%d",&n);
for(;;)
{
if(n==0) break;
cnt=0;
clean(0);
for(int i=1;i<=n;i++)
{
scanf("%s",s[i]);
ans[i].num=0,ans[i].pos=i;
build(s[i],i);
}
ac[0].fail=0;
get_fail();
scanf("%s",q);
ac_query(q);
sort(&ans[1],&ans[n+1],cmp);
printf("%d\n",ans[1].num);
int len=strlen(s[ans[1].pos]);
for(int i=0;i<len;i++) printf("%c",s[ans[1].pos][i]);
puts("");
for(int i=2;i<=n;i++)
{
if(ans[i].num==ans[1].num)
{
int lenth=strlen(s[ans[i].pos]);
for(int j=0;j<lenth;j++) printf("%c",s[ans[i].pos][j]);puts("");
}
else break;
}
scanf("%d",&n);
}
return 0;
}