例题P5496 【模板】回文自动机(PAM)
作用
用 O(n) 的时间复杂度求出一个字符串的所有本质不同的回文子串
原理
(解决方法)
1.自动机节点:在父亲左右各加一个转移边的回文串2号表示a,6号表示aa(如图)
2. 奇根偶根:回文串分奇偶,分情况讨论,1号结点是奇根,0号是偶根(如图)
3.fail(失配数组)
新加入一个字符的时:
我们要从当前节点不断的跳 fail 指针
直到跳到某一个节点所表示的回文串的两侧都能扩展一个待添加的字符
我们就看这个节点有没有这个儿子,如果有就直接走下去,没有就新建一个节点
求每个节点的 fail 指针:
我们显然可以想出一个节点的最长回文后缀
必然是在它父节点的某个回文后缀两侧各拓展一个当前字符得到的
所以新建一个节点之后,我们可以从它父亲的 fail 节点开始,不断的跳 fail 指针
直到跳到第一个两侧能拓展这个字符的节点为止,那么该节点的儿子就是新建节点最长回文后缀
代码(p5496)
#include <bits/stdc++.h>
using namespace std;
char s[500010];
int ans,now,tot=1,len[500010],fail[500010],cnt[500010],sh[500010][28];//结果,上一次的位置,节点数,节点的回文长度,节点的非自身最长回文子串,答案,回文自动机;
int getfail(int u,int id)
{
while(id-len[u]-1<=0||s[id-len[u]-1]!=s[id])//u的回文长度大于id或者不能匹配
u=fail[u];
return u;//返回转移的点
}
int insert(char ss,int id)
{
int fa=getfail(now,id);//树上的转移点
if(!sh[fa][ss-'a'])//没有这个转移边
{
fail[++tot]=sh[getfail(fail[fa],id)][ss-'a'];//失配是不停向上跳
sh[fa][ss-'a']=tot;
len[tot]=len[fa]+2;
cnt[tot]=cnt[fail[tot]]+1;
}
now=sh[fa][ss-'a'];//上一次的位置
return cnt[now];
}
int main()
{
scanf("%s",s+1);
int n=strlen(s+1);
fail[0]=1;len[1]=-1;//初始化:偶数根失配连奇数根,奇数根长度设为-1;
for(int i=1;i<=n;i++)
{
s[i]=(s[i]-'a'+ans)%26+'a';//强制在线
ans=insert(s[i],i);//新增一个点
printf("%d ",ans);
}
return 0;
}