洛谷 P3649 回文树 回文自动机板子

https://www.luogu.org/problemnew/show/P3649

这道题就是一个板子题；

回文自动机这个东西，前几天刚刚听说，看了两天感觉不是怎么好用呢。网上的博客写得也很不错，今天在这里写一个板子。

回文自动机主要就是有两个根，奇数回文串，偶数回文串两个根，fail的和AC自动机差不多，主要注意这些吧。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300010;
struct Palindromic_Tree{
    int nxt[N][30],fail[N],cnt[N];
    int num[N],len[N],s[N];
    int last,n,p;
    int newnode(int l)
    {
        memset(nxt[p],0,sizeof(nxt[p]));
        cnt[p]=num[p]=0;
        len[p]=l;
        return p++;
    }
    void init()
    {
        p=0;
        newnode(0);
        newnode(-1);
        last=n=0;
        s[0]=-1;
        fail[0]=1;
    }
    int get_fail(int x)
    {
        while(s[n-len[x]-1]!=s[n]) x=fail[x];
        return x;
    }
    void add(int c)
    {
        c-='a';
        s[++n]=c;
        int cur=get_fail(last);
        if(!nxt[cur][c])
        {
            int now=newnode(len[cur]+2);
            fail[now]=nxt[get_fail(fail[cur])][c];
            nxt[cur][c]=now;
            num[now]=num[fail[now]]+1;
        }
        last=nxt[cur][c];
        cnt[last]++;
    }
    ll Count()
    {
        ll ret=0;
        for(int i=p-1;i>=0;i--) cnt[fail[i]]+=cnt[i];
        for(int i=0;i<p;i++)
            ret=max(ret,1LL*cnt[i]*len[i]);
        return ret;
    }
}pam;
char str[N];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin>>str;
    int len=strlen(str);
    pam.init();
    for(int i=0;i<len;i++) pam.add(str[i]);
    cout<<pam.Count()<<endl;
    return 0;
}

1楼的评论，主要是fail和初始化有点不一样。

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N = 300010;

struct Palindromic_Tree {
    int nxt[N][30], fail[N], cnt[N];
    int num[N], len[N], s[N];
    int last, n, p;

    int newnode(int l) {
        memset(nxt[p], 0, sizeof(nxt[p]));
        cnt[p] = num[p] = 0;
        len[p] = l;
        return p++;
    }

    void init() {
        p = 0;
        newnode(-1);
        newnode(0);
        n = 0;
        last = 1;
        s[0] = -1;
        fail[1] = 0;
    }

    int get_fail(int x) {
        while (s[n - len[x] - 1] != s[n])
            x = fail[x];
        return x;
    }

    void add(int c) {
        c -= 'a';
        s[++n] = c;
        int cur = get_fail(last);
        if (!nxt[cur][c]) {
            int now = newnode(len[cur] + 2);
            if (nxt[get_fail(fail[cur])][c] != 0)
                fail[now] = nxt[get_fail(fail[cur])][c];
            else fail[now] = 1;
            nxt[cur][c] = now;
            num[now] = num[fail[now]] + 1;
        }
        last = nxt[cur][c];
        cnt[last]++;
    }

    ll Count() {
        ll ret = 0;
        for (int i = p - 1; i >= 0; i--) cnt[fail[i]] += cnt[i];
        for (int i = 0; i < p; i++)
            ret = max(ret, 1LL * cnt[i] * len[i]);
        return ret;
    }
} pam;

char str[N];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> str;
    int len = strlen(str);
    pam.init();
    for (int i = 0; i < len; i++) pam.add(str[i]);
    cout << pam.Count() << endl;
    return 0;
}