[UVA7041 ] The Problem to Slow Down You [回文树]

题目链接
题意：给出两个仅包含小写字符的字符串 A 和 B ；
求：对于 A 中的每个回文子串，B 中和该子串相同的子串个数的总和。

分析：建两颗回文树，然后从０和１节点分别并行dfs扫下去
Code

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

typedef long long ll;
const int N = 200000 + 11;  
const int MAXS = 26 ;  
struct Palindromic_Tree {  
	int next[N][MAXS];//next指针，next指针和字典树类似，指向的串为当前串两端加上同一个字符构成  
	int fail[N] ;     //fail指针，失配后跳转到fail指针指向的节点  
	int cnt[N] ;      //表示(节点i表示的本质不同的串)的个数（建树时求出的不是完全的，最后count()函数跑一遍以后才是正确的） 
	int num[N] ;      //表示以节点i表示的最长回文串的最右端点为回文串结尾的回文串个数
	int len[N] ;	  //len[i]表示节点i表示的回文串的长度（一个节点表示一个回文串）
	int S[N] ;		  //存放添加的字符  
	int last ;		  //指向新添加一个字母后所形成的最长回文串表示的节点。
	int n ;		      //表示添加的字符个数。
	int p ;			  //表示添加的节点个数。

	int newnode ( int l ) {//新建节点  
		for ( int i = 0 ; i < MAXS ; ++ i ) next[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 ( !next[cur][c] ) {       //如果这个回文串没有出现过，说明出现了一个新的本质不同的回文串  
			int now = newnode ( len[cur] + 2 ) ;//新建节点  
			fail[now] = next[get_fail ( fail[cur] )][c] ;//和AC自动机一样建立fail指针，以便失配后跳转  
			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] ;  
		}
		//父亲累加儿子的cnt，因为如果fail[v]=u，则u一定是v的子回文串！  
	}  
}pa, pb;  
char a[N], b[N];
ll ans;
void dfs(int nowa, int nowb){
	for(int i = 0; i < MAXS; i++){
		if(pa.next[nowa][i] && pb.next[nowb][i]){
			ans += pa.cnt[pa.next[nowa][i]] * 1ll * pb.cnt[pb.next[nowb][i]];
			dfs(pa.next[nowa][i], pb.next[nowb][i]);
		}
	}
}
int main(){
	int T; scanf("%d" ,&T);
	for(int cas = 1; cas <= T; cas ++){
		scanf("%s%s", a, b);
		pa.init(); pb.init();
		for(int i = 0; a[i]; i++) pa.add(a[i]);
		for(int i = 0; b[i]; i++) pb.add(b[i]);
		pa.count(); pb.count();
		ans = 0;
		dfs(0, 0);
		dfs(1, 1);
		printf("Case #%d: %lld\n", cas, ans);
	}
	return 0;
}