Gym - 100548G 回文树

想到了建2棵回文树，但想的是利用hash，但会爆内存。

其实直接建出2棵回文树后对跑dfs就行了。

每棵树同时先跑奇偶结点即可。

比如

A树  0结点空

B树  0结点空 

他们都指向aa，所以可以往下dfs  即有相同回文串。

然后发现aa又都指向baab  所以b这里即i=1可以往下dfs

每次算是匹配的回文串贡献即可。

//KX
#include <bits/stdc++.h>
using namespace std;
template<typename T>inline void rd(T&x){
    x=0;int f=1;char ch=getchar();
    while(ch<'0' ||ch>'9'){ if(ch=='-')f=-1; ch=getchar(); }
    while(ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
    x*=f;
}
typedef long long ll;
typedef double db;
#define ls (o<<2)
#define rs (o<<1|1)
#define pb push_back
const int M = 2e5+7 ;
const int N = 27 ;
struct pam{
    int nxt[M][N],fail[M];//在当前状态首尾添加字符的状态,失配跳到的状态
    int len[M],S[M],ppos[M];//状态i表示的回文长度,缓存池,状态对应首次出现的位置
    int cnt[M],num[M];//状态出现次数,以状态末尾结尾但不包含本条路径的数目 
    int last,n,p;//上个状态,总长度,当前状态编号
    int newnode(int l){
        memset(nxt[p],0,sizeof nxt[p]);
        cnt[p]=num[p]=0;
        len[p]=l;
        return p++;
    }
    void init(){
        p=0;
        newnode(0);newnode(-1);
        last=n=0;
        S[n]=-1;
        fail[0]=1;
    }
    int get_fail(int x){
        while(S[n-len[x]-1]!=S[n])x=fail[x];return x;
    }
    void add(int c,int pos){
        c-='a';
        S[++n]=c;
        int cur=get_fail(last);
        if(!nxt[cur][c]){
            int now=newnode(len[cur]+2);
            ppos[now]=pos;
            fail[now]=nxt[get_fail(fail[cur])][c];
            nxt[cur][c]=now;
            num[now]=num[fail[now]]+1;
        }last=nxt[cur][c];cnt[last]++;
    }
    void Count(){
        for(int i=p-1;i>=0;--i)cnt[fail[i]]+=cnt[i];
    }
    //加完所有字符后要跑一次count，把所有状态的字符串的cnt 给补全
}p[2];
//p.p-2本质不同回文串的个数 
char s[M];
ll ans;
void dfs(int x,int y)//两棵回文树的结点分别为x，y 
{
	for(int i=0;i<26;i++)
	{
		int xx=p[0].nxt[x][i];
		int yy=p[1].nxt[y][i];//下一个结点对应的id
		if(xx&&yy)//如果可以继续相同匹配
		{
//			printf("%d  %d  %d\n",i,x,y);
//			printf("--    %d  %d   %d   \n",xx,p[0].ppos[xx],p[0].len[xx]);
//			printf("--    %d  %d   %d   \n",yy,p[1].ppos[yy],p[1].len[yy]);
			ans+=(ll)p[0].cnt[xx]*p[1].cnt[yy];
			dfs(xx,yy);
		}
	}
}
int main()
{
	int t;
	cin>>t;
	for(int ca=1;ca<=t;ca++)
	{
		ans=0;
		scanf("%s",s);
		int len=strlen(s);
		p[0].init();//不要忘了先初始化
		for(int i=0;i<len;i++)
			p[0].add(s[i],i);
			//printf("%d ",p[0].num[p[0].last]);
		//puts("ok");
		p[0].Count();
		scanf("%s",s);
		len=strlen(s);
		p[1].init();//不要忘了先初始化
		for(int i=0;i<len;i++)
			p[1].add(s[i],i);
		p[1].Count();
		dfs(0,0);
		dfs(1,1);
		printf("Case #%d: %lld\n",ca,ans);
	} 
	return 0;
}