2014-2015 ACM-ICPC, Asia Xian Regional Contest - G - The Problem to Slow Down You （ 回文树 ）

题目链接：点击进入

题目

题意

给你两个字符串，让你求相等的回文子串的对数

思路

对两个字符串建两棵回文树，我们知道回文树中 next [ i ] [ j ] : 编号为 i 的回文串两边加上字符 j 所形成的回文串的编号。根据 next 数组，我们可以分别从两棵树的 偶数根 和 奇数根 同时 dfs 找两个字符串相同的回文子串，同时计入答案。

代码

// #pragma GCC optimize("Ofast","inline","-ffast-math")
// #pragma GCC target("avx,sse2,sse3,sse4,mmx")
//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
//#include<bits/stdc++.h>
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<bitset>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
#define max(a,b) a>b?a:b
#define min(a,b) a<b?a:b
//#define int long long
//#define double long double
//#ifndef ONLINE_JUDGE  freopen("data.in.txt","r",stdin);
//freopen("data.out.txt","w",stdout); #endif //文件读取 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int Mod=998244353;
const double eps=1e-9;
const int N=26;
const int mod=51123987;
const int maxn=2e5+10;

/*--------------------------------------------*/ 
inline int read()
{
    int data=0,w=1; char ch=0;
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=data*10+ch-'0',ch=getchar();
    return data*w;
}
/*--------------------------------------------*/


/******邻接矩阵******/
struct Palindromic_Tree 
{
	int next[maxn][N];//next指针，next指针和字典树类似，指向的串为当前串两端加上同一个字符构成
	int fail[maxn];//fail指针，失配后跳转到fail指针指向的节点
	int cnt[maxn];//表示节点i表示的本质不同的串的个数
	int num[maxn];//表示以节点i表示回文串的最右端点为回文串结尾的回文串个数。
	int len[maxn];//len[i]表示节点i表示的回文串的长度
	int S[maxn];//存放添加的字符
	int last;//指向上一个字符所在的节点，方便下一次add
	int n;//字符数组指针
	int p;//节点指针
	int newnode(int l)//新建节点
	{
		for(int i=0;i<N;++i) 
			next[p][i]=0;
		cnt[p]=0;
		num[p]=0;
		len[p]=l;
		return p++;
	}
	void init()//初始化
	{
		p=0;
		newnode(0);//0表示偶数长度串的根,len[0]=0 
		newnode(-1);//1表示奇数长度串的根,len[1]=-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的子回文串！
	}
}p1,p2;

char s1[maxn],s2[maxn];
int n,t,tot;
ll ans;

void dfs(int pos1,int pos2)
{
	for(int i=0;i<26;i++)
	{
		int x1=p1.next[pos1][i];
		int x2=p2.next[pos2][i];
		if(x1&&x2)
		{
			ans+=1LL*p1.cnt[x1]*p2.cnt[x2];
			dfs(x1,x2);
		}
	}
}

int main()																						
{	
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	scanf("%d",&t);
	while(t--)
	{
		scanf("%s",s1);
		scanf("%s",s2);
		p1.init();
		p2.init();
		int l1=strlen(s1);
		int l2=strlen(s2);
		ans=0;
		for(int i=0;i<l1;i++)
			p1.add(s1[i]);
		for(int i=0;i<l2;i++)
			p2.add(s2[i]);
		p1.count();
		p2.count();
		dfs(0,0);
		dfs(1,1);
		printf("Case #%d: %lld\n",++tot,ans);
	}
	return 0;
}