快乐的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);
}