【笔记】回文自动机速学

比赛现场速学。

概念

1. 回文自动机是一个DFA，接受所有回文子串的一半。
2. 它的结构是树，每条边上有一个字符，表示在原节点对应的字符串两端各加一个相同字符，每个节点对应一个回文子串。
3. 回文自动机的结构是两棵树，一棵树中节点对应的回文子串长度都是奇数，另一棵树是偶数。
4. 一个节点的fail指针指向这个节点所代表的回文串的最长回文后缀所对应的节点。
5. 节点需要维护回文子串的长度len。

构造

1. 回文自动机有两个初始状态，分别代表长度为-1和0的回文串，称为奇根和偶根，不代表实际字符串。
2. 偶根的fail指向奇根，奇根没有fail，因为奇根转移出的状态一定是回文子串，不可能失配。
3. 回文自动机同样是增量构造，当添加字符串pos位置的字符c时：
   1. 从以字符串的上一个字符结尾的最长回文子串对应的节点开始，不断沿着fail指针走，直到找到一个节点p满足s[pos-len[p]-1]=c，即p对应回文子串在原串中的上一个字符与待添加字符相同。
   2. 此时，两边添加字符c就是一个新的回文串，如果没有这个节点，就需要新建节点。（注意到，如果一直找不到这样的节点p，就会找到奇根，此时新串就是"c"）。
   3. 新建了节点后需要得到新节点的fail，不断从p跳fail，可以得到cpc的最长回文后缀cbc，如果没有找到，就将其连向偶根。

正确性证明

暂略

模板

struct PalindromeAutomaton
{
    int ch[M][26], fail[M], len[M], cnt[M]; //cnt计算完后需要反向遍历一波
    int sz, lst, s[M], sn; //回文自动机需要把字符串也记录下来
    int create(int id) //初始化第id个节点，适用于多组数据
    {
        memset(ch[id], 0, sizeof(ch[id]));
        fail[id] = len[id] = cnt[id] = 0;
        return sz = id;
    }
    void init() //使用前必须调用init()
    {
        create(0); create(1); 
        fail[0] = 1; len[1] = -1; //奇根和偶根
        lst = sn = 0; s[0] = 26;
    }   
    void extend(char *str)
    {
        for(int i=0; str[i]; ++i)
        {
            int c=str[i] - 'a';
            s[++sn] = c;
            int p=lst; while(s[sn-len[p]-1]!=c) p=fail[p];
            if(!ch[p][c])
            {
                create(sz+1);
                len[sz] = len[p] + 2;
                int tmp = fail[p]; while(s[sn-len[tmp]-1]!=c) tmp=fail[tmp];
                fail[sz] = ch[tmp][c];
                ch[p][c] = sz;
            }
            lst = ch[p][c];
            ++cnt[lst];
        }
        for(int i=sz; i; --i)
            cnt[fail[i]] += cnt[i];
    }
}pam;

洛谷P3649 [APIO2014]回文串

给定字符串s，定义一个子串的存在值为这个子串在s中的出现次数乘以这个子串的长度。
求所有回文子串中的最大存在值。

对字符串s建立回文自动机，现在只需要求每个节点的出现次数。
每当新插入一个字符时，lst以及lst在fail树上的所有祖先出现次数都会加一。
可以不用当场做完，等全部插入完毕后，再反向遍历所有节点，给每个节点的fail的出现次数加上本身的出现次数。

/* LittleFall : Hello! */
#include <bits/stdc++.h>
using namespace std; using ll = long long; inline int read();
const int M = 500016, MOD = 1000000007;

PalindromeAutomaton pam;

char tmp[M];
int main(void)
{
    #ifdef _LITTLEFALL_
    freopen("in.txt","r",stdin);
    #endif

    pam.init();

    scanf("%s", tmp);
    pam.extend(tmp);

    ll ans = 0;
    for(int i=1; i<=pam.sz; ++i)
        ans = max(ans, 1ll * pam.cnt[i] * pam.len[i]);
    cout << ans << endl;


    return 0;
}


inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}