题目大意
求所有回文串出现在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);
}