CodeForces - 17E - Palisection （ 回文树 ）

题目链接：点击进入

题目

题意

对一个给定字符串，求有多少对回文子串有交集

思路

求出这个字符串所有回文子串的对数，然后 所有的 - 无交集的 = 有交集的。
求所有的：
顺序建立回文树，用 sum 数组，通过前缀和，来维护前 i 位总共有多少回文串。
顺序枚举每个位置，以当前位置为末尾的回文串个数 cnt 乘 当前位置之前的回文串个数 sum [ i - 1 ] ，加上当前位置自身的贡献 cnt * ( cnt -1 ) / 2 ，累加答案 ans1 。
求无交集的：
逆序建立回文树，逆序枚举每个位置，对于每个位置 i ,无交集的方案数 = 以这个位置为结尾形成的回文串数 * sum [ i - 1 ] ( 此位置左边所有的回文串数 ),累加答案 ans2。
最后两者做差 ( ans1 - ans2 )，就是所求有交集的答案。
因为此题卡内存比较严，所以可以用邻接表实现回文树，用取模维护前缀和（ 用 define int long long 也会卡 ） 。

代码

// #pragma GCC optimize("Ofast","inline","-ffast-math")
// #pragma GCC target("avx,sse2,sse3,sse4,mmx")
//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
//#include<bits/stdc++.h>
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<bitset>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
#define max(a,b) a>b?a:b
#define min(a,b) a<b?a:b
//#define int long long
//#define double long double
//#ifndef ONLINE_JUDGE  freopen("data.in.txt","r",stdin);
//freopen("data.out.txt","w",stdout); #endif //文件读取 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int Mod=998244353;
const double eps=1e-9;
const int N=1e6+10;
const int mod=51123987;
const int maxn=2e6+10;

/*--------------------------------------------*/ 
inline int read()
{
    int data=0,w=1; char ch=0;
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=data*10+ch-'0',ch=getchar();
    return data*w;
}
/*--------------------------------------------*/


/*邻接表*/
struct Palindromic_Tree 
{
    vector<pair<int,int>>next[maxn];//next指针，next指针和字典树类似，指向的串为当前串两端加上同一个字符构成
	int fail[maxn];//fail指针，失配后跳转到fail指针指向的节点
	int cnt[maxn];//表示节点i表示的回文串出现的次数
	int num[maxn];//表示以节点i表示回文串的最右端点为回文串结尾的回文串个数。
	int len[maxn];//len[i]表示节点i表示的回文串的长度
	int S[maxn];//存放添加的字符
	int last;//指向上一个字符所在的节点，方便下一次add
	int n;//字符数组指针
	int p;//节点指针
    int newnode(int w)
    {
        next[p].clear();
        cnt[p]=num[p]=0;
        len[p]=w;
        return p++;
    }
    void init()
    {
        p=0;
        newnode(0);
        newnode(-1);
        last=0;
        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)
    {
        c-='a';
        S[++n]=c;
        int cur=get_fail(last);
        int flag=0;
        for(int i=0;i<next[cur].size();++i)
        {
        	if(next[cur][i].first==c)
            {
                last=next[cur][i].second;
                cnt[last]++;
                return ;
            }
		}
        int now=newnode(len[cur]+2);
        int fi=get_fail(fail[cur]);
        flag=0;
        for(int i=0;i<next[fi].size();i++)
        {
        	if(next[fi][i].first==c)
            {
                flag=next[fi][i].second;
                break;
            }
		}
        fail[now]=flag;
        next[cur].push_back(make_pair(c,now));
        num[now]=num[flag]+1;
        last=now;
        cnt[now]++;
    }
    void count()
    {
        for(int i=p-1;i>1;--i)
            cnt[fail[i]] += cnt[i];
    }
}patree;

char str[maxn];
int n;
int sum[maxn];

int main()																						
{	
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	scanf("%d",&n);
	scanf("%s",str);
	ll ans=0;
	patree.init();
	for(int i=0;i<n;i++)
	{
		patree.add(str[i]);
		int pos=patree.last;
		ans=(ans+1LL*patree.num[pos]*sum[i-1]+1LL*patree.num[pos]*(patree.num[pos]-1)/2)%mod;
		sum[i]=(sum[i-1]+patree.num[pos])%mod;
	}
	patree.init();
	for(int i=n-1;i>0;i--)
	{
		patree.add(str[i]);
		int pos=patree.last;
		ans=(ans-1LL*patree.num[pos]*sum[i-1]%mod+mod)%mod;
	}
    printf("%lld",ans);
	return 0;
}