首页 > 其他分享 >回文自动机(PAM) 详解

回文自动机(PAM) 详解

时间:2023-10-01 21:00:11浏览次数:39  
标签:int len 后缀 Fail 自动机 PAM 回文

PAM 是一种高效存储字符串中所有回文子串的自动机,用于解决回文串相关问题。
虽然代码稍微长一点,但写起来比 manacher 容易很多,毕竟没有加了一堆字符再转回原串的若干上取整下取整问题。

前置知识

无。或许需要一些自动机相关的理论基础。

结构 & 定义

状态

我们用 PAM 上的一个节点来表示一个回文子串,作为 PAM 的一个状态。但回文串分奇偶两种,像 manacher 一样在每两个字符之间加分隔符是很麻烦的。因此,我们把 PAM 的状态分成两个部分,一部分存奇回文串,另一部分存偶回文串。

同理,我们把根也分为奇根和偶根。它们不表示任何字符串,只作为初始状态而存在。
因为存的是回文串,我们其实只需要对一个串记录其中一半位置的字符是什么,所以定义 PAM 上的一个点到根的路径上的字符串表示它所代表的回文串的其中一半,这一点上 PAM 与以前学过的自动机状态的定义都不同。
换句话说,对于一个点它实际表示的回文串,在 PAM 上的读法是从它开始沿着 PAM 读到根,再原路读回该点形成的字符串。这里注意如果是奇回文串,与根相连的那个字符边只读一次。

令奇根为 \(1\),偶根为 \(0\)。那么对于 \(s=\texttt{"abaabc"}\),建出的 PAM 如下图(省略 Fail 指针):

每个点表示的回文串根据上文所述即可读出,例如点 \(7\) 表示 \(\texttt{c}\),点 \(4\) 表示串 \(\texttt{aba}\),点 \(6\) 表示串 \(\texttt{baab}\)。

Fail 指针

在 PAM 中,每个节点同样有一个 Fail 指针。这里 \(\text{Fail}(x)\) 的定义是 \(x\) 表示的回文串的最长回文后缀的状态。(也同时是最长回文前缀,因为 \(x\) 是回文串)
对于初始状态,我们定义偶根的 Fail 指向奇根。而我们并不关心奇根的 Fail,因为任意一个长度为 \(1\) 的字符串都是回文串,奇根不可能失配。

同时,对于每个节点我们记录 \(\text{len}(x)\) 表示它实际代表的回文串长度,用于在下文 PAM 的构造中判断每个点在原串中的位置。那么对于点 \(x\),设它在 PAM 上的父亲是 \(fa(x)\),则有 \(\text{len}(x)=\text{len}(fa(x))+2\)。
为保证奇回文串也满足这个性质,令 \(\text{len}(1)=-1\)。

对于之前的例子,建出它的 Fail 指针就是这样的(为了减少交叉,把 \(2\) 和 \(7\) 号点换了位置):

PAM 的构造

PAM 的构造方式是在线的,即每次添加一个字符 \(c\) 时,在原来的 PAM 基础上添加与新增字符相关的状态和转移。
假设对于一个长度为 \(n\) 的字符串 \(s\),我们已经构造完了前 \(n-1\) 个字符,现在要加入第 \(n\) 个字符。设这个字符为 \(c\),前 \(n-1\) 个字符最长回文后缀对应的状态是 \(now\)。
要新增的状态就是以第 \(n\) 个字符结尾,且在以前没有出现过的回文串。考虑到一个回文串前后各去掉一位还是回文串,新增的回文串前后去掉一位,一定是某个以 \(n-1\) 为结尾的回文串。
我们要找的就是以 \(n-1\) 为结尾的回文串中,前一位恰好是 \(c\) 的最长的串。发现上一次构造到的节点 \(now\) 就表示以 \(n-1\) 为结尾的最长回文串,因此我们不断令 \(now\gets \text{Fail}(now)\),直至满足条件。设使条件满足的状态为 \(pos\)。那么以 \(n\) 为结尾的最长回文后缀就是在 \(x\) 前后各加一个 \(c\)。根据 PAM 的定义,这个状态就是 \(pos\) 通过字符 \(c\) 的边连向的儿子。

如果 \(pos\) 没有对应的儿子,代表这个回文串是新增的,我们添加一个新的状态。否则既然已经存在就不用管了。
那么对于这个新的状态,可以证明只有最长的这个回文后缀是新增的。考虑一个比它短的回文后缀,可以在最长的里面对称到一个不包括 \(n\) 的字符串,这一定是以前出现过的状态。

(如果黑的是最长回文后缀,蓝的是次长回文后缀,那显然红的和蓝的相同,也是回文串,在前面出现过。)

那么我们令新点为 \(new\),容易得出 \(fa(new)=pos\),\(\text{len}(new)=\text{len}(pos)+2\)。
只剩下它的 Fail 还没有求了。

发现求 Fail 指针和求最长以 \(n\) 结尾的回文后缀的本质是一样的。这次我们从 \(\text{Fail}(pos)\) 开始跳,找到第一个能在前后各加一个 \(c\) 的回文串即可。
如果到最后都没匹配到,令 \(\text{Fail}(new)=0\) 就好了。
你可能会奇怪为什么不指向奇根呢,这两个都是啥也没有。考虑这样的情况:原来字符串只有一个字符 \(\texttt{a}\),现在变成了 \(\texttt{aa}\)。那么这个新回文串是从偶根同时也是之前的 Fail 转移过来的。如果一开始把 \(a\) 的 Fail 指向奇根就会转移不到这种情况。

看起来构建完了。

实现

结构体定义是平凡的:

struct node{int len,fa,s[26];} d[N];

设函数 getfail(u,i) 表示从状态 \(u\) 开始查找首个前一位与 \(s_i\) 相同的回文后缀状态。

il int getfail(int u,int i)
{
    while(i-d[u].len-1<=0||s[i-d[u].len-1]!=s[i]) u=d[u].fa;
    return u;
}

具体的构造根据上文也不难写出。

d[0].fa=1,d[1].len=-1;
for(int i=1,now=0;i<=n;i++)
{
    int pos=getfail(now,i),c=s[i]-'a';
    if(!d[pos].s[c])
    {
        d[++tot].fa=d[getfail(d[pos].fa,i)].s[c];
        d[pos].s[c]=tot,d[tot].len=d[pos].len+2;
    }
    now=d[pos].s[c];
}

注意一定要先对新点求 Fail,再连新的转移边。
考虑颠倒顺序会出什么问题,上文说过“最后匹配不到,要令 Fail=0”,正常情况下匹配不到,\(pos\) 就没有 \(c\) 的转移边,Fail 确实是 \(0\);而如果先给 \(pos\) 连了新节点的边,最后匹配不到,新节点就会把 \(Fail\) 连到自己,喜提死循环。

看起来也实现完了。

正确性证明

接下来对 PAM 的时空复杂度均为线性给出证明。(因为这个还比较好证,那就写一下。

状态数证明

PAM 的状态数即为字符串本质不同的回文子串个数。
考虑每次新增一个字符,上文已经证明过只有以它结尾的最长回文后缀可能是新增的。也就是说,每次新增字符,本质不同的回文子串数至多增加 \(1\)。
因此任意字符串本质不同回文子串数不多于 \(n\),上界可以在形如 \(s=\texttt{"aaaa}\dots\texttt{aaaa"}\) 的字符串中取到。

时间复杂度证明

看起来除了我们每次匹配都在循环跳 Fail 指针以外,线性复杂度都很显然。
考虑 \(now\) 在 Fail 树上的深度,每次新增一个字符至多使它的深度增加 \(1\)。那么 \(n\) 个字符至多增加了 \(n\) 次,\(now\) 至多跳了 \(2n\) 次 Fail。
而求 Fail 指针时的那个循环也是同理的。因此 PAM 的时间复杂度是线性。

例题

P5496【模板】回文自动机(PAM)
根据 Fail 的定义,发现对于 PAM 上的一个状态,以它结尾的回文后缀个数就是它在 Fail 树上的深度。
那么我们只需要在构建过程中记录每个状态在 Fail 树上的深度即可。

点击查看代码
const int N=5e5+5;
int n,tot=1;
struct node {int len,fa,dep,s[26];}d[N];
char s[N];
il int getfail(int u,int i)
{
    while(i-d[u].len-1<=0||s[i-d[u].len-1]!=s[i]) u=d[u].fa;
    return u;
}
int main()
{
    scanf("%s",s+1); n=strlen(s+1);
    d[0].fa=1,d[1].len=-1;
    int now=0,lst=0;
    for(int i=1;i<=n;i++)
    {
        s[i]=(s[i]-'a'+lst)%26+'a';
        int pos=getfail(now,i); int c=s[i]-'a';
        if(!d[pos].s[c])
        {
            d[++tot].fa=d[getfail(d[pos].fa,i)].s[c];
            d[pos].s[c]=tot,d[tot].len=d[pos].len+2;
            d[tot].dep=d[d[tot].fa].dep+1;
        }
        now=d[pos].s[c];
        printf("%d ",lst=d[now].dep);
    }
    return 0;
}

P3649 [APIO2014] 回文串

这题要记录每个回文串出现的次数。在 Fail 树的角度考虑,每加入一个字符,最长回文后缀所在的状态在 Fail 树上的所有祖先出现次数都 \(+1\)。
显然每次都跳 Fail 暴力修改是不可行的,因此只在最长回文后缀处打一个标记,最后统计答案时,状态的出现次数就是 Fail 树的子树和。

点击查看代码
const int N=3e5+5;
int n,tot=1;
long long f[N];
char s[N];
struct node{int len,fa,s[26];} d[N];
il int getfail(int u,int i)
{
    while(i-d[u].len-1<=0||s[i-d[u].len-1]!=s[i]) u=d[u].fa;
    return u;
}
int main()
{
    scanf("%s",s+1); n=strlen(s+1);
    d[0].fa=1,d[1].len=-1;
    for(int i=1,now=0;i<=n;i++)
    {
        int pos=getfail(now,i),c=s[i]-'a';
        if(!d[pos].s[c])
        {
            d[++tot].fa=d[getfail(d[pos].fa,i)].s[c];
            d[pos].s[c]=tot,d[tot].len=d[pos].len+2;
        }
        now=d[pos].s[c],f[now]++;
    }
    long long ans=0;
    for(int i=tot;i;i--)
    {
        f[d[i].fa]+=f[i];
        ans=max(ans,1ll*f[i]*d[i].len);
    }
    printf("%lld\n",ans);
    return 0;
}

P4199 万径人踪灭
ybtoj 翻到的。其实这题重点不在 PAM/manacher,倒不如说是 FFT 练习题(
答案可以转化成所有合法子序列减掉连续的。
对于不保证连续的,发现本质是求下标和等于某个数(即关于某个点对称)且相同的点对数。这个东西形如多项式卷积,对两种字符分别 FFT。
而连续回文子串的总数可以用 PAM 来求,代码不放了。

标签:int,len,后缀,Fail,自动机,PAM,回文
From: https://www.cnblogs.com/ying-xue/p/17738822.html

相关文章

  • PAM+回文串border理论
    PAM板子#include<bits/stdc++.h>usingnamespacestd;constintmaxn=300000+5;namespacepam{ intsz,tot,last; intcnt[maxn],ch[maxn][26],len[maxn],fail[maxn]; chars[maxn]; intnode(intl){//建立一个新节点,长度为l sz++; memset(ch[sz],......
  • 5. 最长回文子串
    https://leetcode.cn/problems/longest-palindromic-substring/description/前置知识取字符串的子串s.substr(start,len);//start是子串的起始位置,从0开始,len是子串的长度思路枚举回文字符串的中间位置i,如果回文字符串的长度为奇数,则从左右i-1,i+1两个方向遍历字符串,判断是否满足......
  • 力扣刷题笔记-05 最长回文子串
    05最长回文子串半山腰有点拥挤,你要去山顶看看。中心扩展法什么是回文从左边出发,字符的顺序和从右边出发是一样的,比如aba,abba。那么基于这个理论,我们就可以想到解决方案:找一个中心点,向两边出发,左右两边各移动一位,如果相同就证明是回文子串,不相同就停止,找下一个中心点中心点......
  • Aho-Corasick 算法 AC自动机实现
    敏感词过滤在社区发帖、网站检索、短信发送等场景下是很常见的需求,尤其是在高并发场景下如何实现敏感词过滤,都对过滤算法提出了更高的性能要求,Ahocorasick算法能够实现毫秒级的万字过滤匹配,能够很好的满足各种场景下的敏感词过滤需求。Aho-Corasick算法通过将模式串预处理为确定......
  • Death DBMS题解(AC自动机)
    题目传送门CF1437G好题观察这道题,发现有关字串的题目,一般来说,这种题都要构建\(AC\)自动机,所以考虑构建。构建之后,原来的所有\(fail\)是一个树形结构。解法\(1\):考虑从询问入手,那么对于每一个询问,等价于就是查询每一个\(Q_i\)包含的后缀的最大值,再对这些最大值取\(max......
  • Manacher——最快的找最长回文算法
    Manacher马拉车——Manacher算法解决的问题给定一串字符串str,求str内的最长回文子串,我们可以从最朴素的算法开始,逐渐深入Manacher算法。朴素穷举法一直枚举字符串str的子串,并判断子串是否为回文。这个时间复杂度直接到\(O(n^3)\)了,一般题目都会超时。中心扩散法作为一个回文......
  • 最长回文子串
    给你一个字符串 s,找到 s 中最长的回文子串。如果字符串的反序与原始字符串相同,则该字符串称为回文字符串。 示例1:输入:s="babad"输出:"bab"解释:"aba"同样是符合题意的答案。示例2:输入:s="cbbd"输出:"bb" classSolution{public:stringlongestPalindrome(strings......
  • 1275MPaMJ螺纹螺栓
    声明本文是学习GB-T42850-20231275MPaMJ螺纹螺栓.而整理的学习笔记,分享出来希望更多人受益,如果存在侵权请及时联系我们1范围本文件规定了1275MPaMJ螺纹螺栓(以下简称螺栓)的结构、尺寸和技术要求。本文件适用于1275MPaMJ螺纹螺栓的设计、制造和验收。2规范性引用文件......
  • 用栈来判断回文数
    代码如下所示:/***用栈实现回文数**/publicclassTest2{publicstaticvoidmain(String[]args){//测试代码System.out.println(isPalindromic("1221"));//输出true}//判断字符串是否回文,如果是数字的话可以转换为字符串再进行回文......
  • mpam linux kernel源码分析
    MPAM(MemorySystemResourcePartitioningandMonitoring)是Armv8.4的feature,用于cache和内存带宽的监控和限制。截至现在,该feature在linuxkernel的实现还在推进,最新一版参见https://git.kernel.org/pub/scm/linux/kernel/git/morse/linux.git/log/?h=mpam/snapshot/v6.5-rc1。......