【模板】AC自动机(洛谷3808&洛谷3796)
清明时节雨纷纷,省选只能当路人。
最近在狂补算法,AC自动机还是比较难理解。找到了洛谷日报上比较容易理解的版本,才勉强理解。
https://www.luogu.org/blog/42196/qiang-shi-tu-xie-ac-zi-dong-ji
千万记得前置技能trie树和KMP!
我博客里也有有关KMP的感想。(顺带安利一发
https://www.cnblogs.com/NightRaven/p/10554401.html
至于trie树,是个比较简单(应该)的数据结构,稍微看看就会了8(大概)(高端操作依然不会)
————————————————
总之以下是洛谷上简单版的AC自动机(加强版请再往下翻翻
#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=n;i++)
#define pb push_back
using namespace std;
const int maxn=1000005;
queue<int> q;
struct ac_automation
{
int tr[maxn][30],cnt,fail[maxn],val[maxn];
void clear_ac()
{
memset(fail,0,sizeof(fail));
memset(val,0,sizeof(val));
cnt=0;
}
void add(char *s)
{
int len=strlen(s),k=0;
rep(i,0,len-1)
{
int c=s[i]-'a';
if(!tr[k][c]) tr[k][c]=++cnt;
k=tr[k][c];
}
val[k]++;
}
void built()
{
rep(i,0,25) if(tr[0][i]) q.push(tr[0][i]);
while(!q.empty())
{
int x=q.front();
q.pop();
rep(i,0,25) if(tr[x][i])
fail[tr[x][i]]=tr[fail[x]][i],q.push(tr[x][i]);
else tr[x][i]=tr[fail[x]][i];
}
}
int query(char *s)
{
int len=strlen(s),k=0,ans=0;
rep(i,0,len-1)
{
int c=s[i]-'a';
k=tr[k][c];
for(int j=k;j && val[j]!=-1;j=fail[j]) ans+=val[j],val[j]=-1;//记得看题,每个词只记一次,若多次就不用赋值为-1.
}
return ans;
}
} acauto;
int main()
{
int n;
scanf("%d",&n);
acauto.clear_ac();
rep(i,1,n)
{
char c[maxn];
scanf("%s",c);
acauto.add(c);
}
char p[maxn];
scanf("%s",p);
acauto.built();
printf("%d",acauto.query(p));
return 0;
}
下面是加强版
#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=n;i++)
#define pb push_back
using namespace std;
const int maxn=1000005;
queue<int> q;
char c[155][75];
int ans[155];
struct ac_automation
{
int tr[maxn][30],cnt,fail[maxn],val[maxn];
void clear_ac()
{
memset(fail,0,sizeof(fail));
memset(val,0,sizeof(val));
memset(tr,0,sizeof(tr));
cnt=0;
}
void add(char *s,int j)
{
int len=strlen(s),k=0;
rep(i,0,len-1)
{
int c=s[i]-'a';
if(!tr[k][c]) tr[k][c]=++cnt;
k=tr[k][c];
}
val[k]=j;
}
void built()
{
rep(i,0,25) if(tr[0][i]) q.push(tr[0][i]),fail[tr[0][i]]=0;
while(!q.empty())
{
int x=q.front();
q.pop();
rep(i,0,25) if(tr[x][i])
fail[tr[x][i]]=tr[fail[x]][i],q.push(tr[x][i]);
else tr[x][i]=tr[fail[x]][i];
}
}
void query(char *s)
{
int len=strlen(s),k=0;
rep(i,0,len-1)
{
int c=s[i]-'a';
k=tr[k][c];
for(int j=k;j;j=fail[j]) ans[val[j]]++;
}
}
} acauto;
int main()
{
int n;
scanf("%d",&n);
while(n)
{
memset(ans,0,sizeof(ans));
acauto.clear_ac();
rep(i,1,n)
{
scanf("%s",c[i]);
acauto.add(c[i],i);
}
char p[maxn];
scanf("%s",p);
acauto.built();acauto.query(p);
int maxo=-1;
rep(i,1,n) maxo=max(maxo,ans[i]);
printf("%d\n",maxo);
rep(i,1,n) if(ans[i]==maxo) printf("%s\n",c[i]);
scanf("%d",&n);
}
return 0;
}
最近严重意识到自己菜的真实,很多算法都不会,又有很多的忘记了,DP也菜的不行。。。最近加油吧。