快乐的JYY

题目大意

求所有回文串出现在A、B两个字符串的次数的积的和。

裸题

做完A后把last变回0，然后再插入B。
每个结点有两个size分别表示在A和B出现的次数，然后就可以统计。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const ll maxn=100000+10;
ll pre[maxn],len[maxn],size[maxn][2],g[maxn][26];
ll i,j,k,l,t,n,m,ans,tot,last;
char s[maxn];
void insert(char ch,int p){
 ll x=last;
 while (s[i-len[x]-1]!=s[i]) x=pre[x];
 if (!g[x][ch-'A']){
 len[++tot]=len[x]+2;
 ll y=pre[x];
 while (s[i-len[y]-1]!=s[i]) y=pre[y];
 pre[tot]=g[y][ch-'A'];
 g[x][ch-'A']=tot;
 }
 x=g[x][ch-'A'];
 size[x][p]++;
 last=x;
}
int main(){
 scanf("%s",s+1);
 n=strlen(s+1);
 len[pre[last=0]=tot=1]=-1;
 pre[1]=1;
 fo(i,1,n) insert(s[i],0);
 scanf("%s",s+1);
 n=strlen(s+1);
 last=0;
 fo(i,1,n) insert(s[i],1);
 fd(i,tot,2) 
 fo(j,0,1)
 if (pre[i]>1) size[pre[i]][j]+=size[i][j];
 fo(i,2,tot) ans+=size[i][0]*size[i][1];
 printf("%lld\n",ans);
}