参考代码:http://blog.csdn.net/gatevin/article/details/44783041
首先要知道,一个长度为n的字符串最多有n个本质不同的回文串
所以回文树中最多有n个节点。
回文树:把本质不同的回文串用fail和nxt数组联系起来
const int maxn = 300000+10;
const int N = 26;
struct PT{
int nxt[maxn][N],fail[maxn];
int p,last,cnt[maxn],len[maxn];//p:添加的节点个数 n:添加的字符个数 last:以新添加的字符结尾的最长字符串在树中节点编号
int n,S[maxn];//S]i]:第i个字符 cnt[i]:第i个节点的字符串个数
int newnode(int l){
for(int i=0;i<N;i++) nxt[p][i]=0;
cnt[p]=0; len[p]=l;
return p++;
}
void init(){
last=p=n=0;
S[0]=-1,fail[0]=1;//fail[0]赋值为1,不能赋值为0
newnode(0),newnode(-1);//先建立0节点,再建立-1节点!!!
}
int get_fail(int x){
while(S[n]!=S[n-len[x]-1]) x=fail[x];
return x;
}
void add(int c){
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];//求now节点的最长回文后缀 cur是它除去两侧‘c’分符后的字符串编号
nxt[cur][c]=now;
}
last=nxt[cur][c];
cnt[last]++;
}
void count(){
for(int i=p-1;i>=0;i--) cnt[fail[i]]+=cnt[i];
}
};