类似AC自动机,树里面每个节点代表主串的一个回文子串,且每个回文子串都不同
主要功能:
1.求串S前缀0~i内本质不同回文串的个数(两个串长度不同或者长度相同且至少有一个字符不同便是本质不同)
2.求串S内每一个本质不同回文串出现的次数
3.求串S内回文串的个数
4.求以下标i结尾的回文串的个数
参考链接:
https://blog.csdn.net/yxuanwkeith/article/details/65667401
https://blog.csdn.net/qq_38995588/article/details/80644655
const int MAXN=2000000+10;
const int NUM=26;
const char C='a';
struct Pal_Tree //回文树
{
int next[MAXN][NUM];
int fail[MAXN];
int cnt[MAXN];
int num[MAXN];
int len[MAXN];
int S[MAXN];
int last,n,p;
int newnode(int l)
{
for(int i=0;i<NUM;i++) next[p][i]=0;
cnt[p]=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)
{
while(S[n-len[x]-1]!=S[n]) x=fail[x];
return x;
}
void add(int c)
{
c-=C;
S[++n]=c;
int cur=get_fail(last);
if(!next[cur][c])
{
int now=newnode(len[cur]+2);
fail[now]=next[get_fail(fail[cur])][c];
next[cur][c]=now;
num[now]=num[fail[now]]+1;
}
last=next[cur][c];
cnt[last]++;
}
void count()
{
for(int i=p-1;i>=0;i--) cnt[fail[i]]+=cnt[i];
}
};