Tsinsen A1393. Palisection

题面

题意

给出一个字符串,求其中相交的回文串的对数.

做法

可以反过来考虑,求出不相交的所有回文串对数,最后再用总数减去.
首先可以建一个回文自动机,求出再i点左边有几个回文串,然后再将字符串倒过来建一个回文自动机,求出以每个点为左端点的回文串的个数,将上下两个量分别相乘即可求出不相交的回文串的个数.

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
#define N 2001000
#define M 51123987
using namespace std;

ll n,last,tt=1;
ll qz[N],ans;
char str[N];
struct Node
{
    int son[26],fail,num,len;
    void init()
    {
        memset(son,0,sizeof(son));
        num=fail=len=0;
    }
}node[N];

inline void add(ll u)
{
    int p,v=str[u]-'a';
    for(;str[u]!=str[u-node[last].len-1];last=node[last].fail);
    if(!node[last].son[v])
    {
        node[++tt].len=node[last].len+2;
        for(p=node[last].fail;str[u]!=str[u-node[p].len-1];p=node[p].fail);
        node[tt].fail=node[p].son[v];
        node[last].son[v]=tt;
        node[tt].num=node[node[tt].fail].num+1;
    }
    last=node[last].son[v];
}

int main()
{
    int i,j;
    cin>>n;
    scanf("%s",str+1);
    node[0].fail=1;
    node[1].len=-1;
    for(i=1;i<=n;i++)
    {
        add(i);
        qz[i]=(qz[i-1]+node[last].num)%M;
    }
    ans=qz[n]*(qz[n]-1)/2%M;
    for(i=0;i<=tt;i++) node[i].init();
    node[0].fail=1;
    node[1].len=-1;
    last=0,tt=1;
    for(i=1;i<=n/2;i++)
    {
        swap(str[i],str[n-i+1]);
    }
    for(i=1;i<n;i++)
    {
        add(i);
        ans=(ans+M-node[last].num*qz[n-i]%M)%M;
    }
    cout<<ans;
}