求字符串中回文子串的个数（回文树详解）

 

写法一： 

#include
#include
#include
#include
#include
#include
#include
#include

写法二：

#include 
#include 
#include 
#include 
#define ri register int
using namespace std;

const int maxn = 300010;
int n, m, tot, cnt[maxn], len[maxn], fail[maxn], last, son[maxn][27], cur;
char s[maxn];
long long ans;
//添加节点 
inline int new_node(ri x)
{
	//更新len，cnt 
	len[tot] = x; cnt[tot] = 0;
	return tot++;
}
//找到最长后缀节点 
inline int get_fail(ri x, ri n)
{
	while(s[n-len[x]-1] != s[n]) x = fail[x];
	return x;
}
//
inline void init()
{
	scanf("%s", s+1);
	n = strlen(s+1);
	//初始化两个根节点 
	new_node(0); new_node(-1);
	fail[0] = 1; last = 0;
}

int main()
{
	init();
	for(ri i=1;i<=n;i++){
		ri x = s[i] - 'a';
		cur = get_fail(last, i);//找到最长后缀回文串 
		if(!son[cur][x]){//没有此回文串 
			ri nw = new_node(len[cur]+2);
			fail[nw] = son[get_fail(fail[cur],i)][x];
			son[cur][x] = nw;
		}
		last = son[cur][x];
		cnt[last]++;
	}
	//计算出来的cnt[i]表示i节点回文子串出现的次数 
	for(int i=tot-1;i>=0;i--) cnt[fail[i]] += cnt[i];
	for(int i=2;i