UVALive - 7041 The Problem to Slow Down You （回文树）

https://vjudge.net/problem/UVALive-7041

题意

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

分析

从0和1两个根节点DFS下去，如果两个相同的节点同时存在就统计答案。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <set>
#include <bitset>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(a, b) memset(a, b, sizeof(a))
#define pb push_back
#define mp make_pair
#define pii pair<int, int>
//#define eps 0.0000000001
#define IOS ios::sync_with_stdio(0);cin.tie(0);
#define random(a, b) rand()*rand()%(b-a+1)+a
#define pi acos(-1)
//const ll INF = 0x3f3f3f3f3f3f3f3fll;
const int inf = 0x3f3f3f3f;
const int maxn = 2e5 + ;
const int maxm =  +;
const int mod = ;

struct PAM{
    int nxt[maxn][];
    int fail[maxn];
    int cnt[maxn];
    int num[maxn];
    int len[maxn];
    int s[maxn];
    int last,n,p;

    int newnode(int w){
        for(int i=;i<;i++) nxt[p][i]=;
        num[p]=cnt[p]=;
        len[p]=w;
        return p++;
    }
    void init(){
        n=last=p=;
        newnode();
        newnode(-);
        s[n]=-;
        fail[]=;
    }
    int get_fail(int x){
        while(s[n-len[x]-]!=s[n]) x=fail[x];
        return x;
    }
    void add(int c){
        c-='a';
        s[++n]=c;
        int cur=get_fail(last);
        if(!nxt[cur][c]){
            int now=newnode(len[cur]+);
            fail[now]=nxt[get_fail(fail[cur])][c];
            nxt[cur][c]=now;
            num[now]=num[fail[now]]+;
        }
        last=nxt[cur][c];
        cnt[last]++;
    }
    void Count(){
        for(int i=p-;i>=;i--) cnt[fail[i]]+=cnt[i];
    }
};
PAM pam1,pam2;
char a[maxn],b[maxn];
ll dfs(int an,int bn){
    ll res=;
    for(int i=;i<;i++)
        if(pam1.nxt[an][i]!=&&pam2.nxt[bn][i]!=){
            res+=(ll)pam1.cnt[pam1.nxt[an][i]]*pam2.cnt[pam2.nxt[bn][i]]+dfs(pam1.nxt[an][i],pam2.nxt[bn][i]);
        }
    return res;
}
int main() {
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
//    freopen("input.txt", "w", stdout);
#endif
    int T,cas=;
    scanf("%d",&T);

    while(T--){
        pam1.init();
        pam2.init();
        scanf("%s%s",a,b);
        int l1=strlen(a),l2=strlen(b);
        for(int i=;i<l1;i++) pam1.add(a[i]);
        for(int i=;i<l2;i++) pam2.add(b[i]);
        pam1.Count();
        pam2.Count();
        ll ans=dfs(,)+dfs(,);
        printf("Case #%d: %lld\n",cas++,ans);
    }
    return ;
}