回文树板子

#include<bits/stdc++.h>
using namespace std;

#define ll long long
#define ull unsigned long long
const int maxn = 300005 ;
const int N = 26 ;
int nxt[maxn][N] ;//next指针，next指针和字典树类似，指向的串为当前串两端加上同一个字符构成
int fail[maxn] ;//fail指针，失配后跳转到fail指针指向的节点
int cnt[maxn] ;
int num[maxn] ;
int len[maxn] ;//len[i]表示节点i表示的回文串的长度
int S[maxn] ;//存放添加的字符
int last ;//指向上一个字符所在的节点，方便下一次add
int n ;//字符数组指针
int p ;//节点指针
char a[maxn];
int id[maxn];

int newnode ( int l )  //新建节点
{
    for ( int i = 0 ; i < N ; ++ i ) nxt[p][i] = 0 ;
    cnt[p] = 0 ;
    num[p] = 0 ;
    len[p] = l ;
    return p ++ ;
}

void init ()  //初始化
{
    p = 0 ;
    newnode (  0 ) ;
    newnode ( -1 ) ;
    last = 0 ;
    n = 0 ;
    S[n] = -1 ;//开头放一个字符集中没有的字符，减少特判
    fail[0] = 1 ;
}

int get_fail ( int x )  //和KMP一样，失配后找一个尽量最长的
{
    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] ;//和AC自动机一样建立fail指针，以便失配后跳转
        nxt[cur][c] = now ;
        num[now] = num[fail[now]] + 1 ;
    }
    last = nxt[cur][c] ;
    cnt[last] ++ ;
    id[last]=n;
}


void count ()
{
    for ( int i = p - 1 ; i >= 0 ; -- i ) cnt[fail[i]] += cnt[i] ;
    //父亲累加儿子的cnt，因为如果fail[v]=u，则u一定是v的子回文串！
    
}
int main()
{
    scanf("%s",a+1);
    int len1=strlen(a+1);
    init();
    for(int i=1; i<=len1; i++)
        add(a[i]);
    count();
    for(int i=2; i<p; i++) // 打印所有回文串及其出现次数 
    {
        for(int j=id[i]-len[i]+1; j<=id[i]; j++)
            printf("%c",a[j]);
        printf(" cnt=%d",cnt[i]);
        printf("\n");
    }
}