Codeforces Beta Round #17 E - Palisection 回文树写法

这一题很卡空间，如果用回文数的话。

我们可以用vector代替  next数组。

省20倍空间大概。

直接算相交回文串很难。

但算不相交回文串很简单。

我们知道，用回文树，能维护出以i结尾的回文串的个数。

再倒着维护一边字符串，就知道以i开头的回文串的个数。

然后记录前缀和直接算即可。

全部的回文串个数更简单。。用cnt数组或者直接用我们的前缀和也行  可以节省空间。

 

//KX
#include<bits/stdc++.h>
using namespace std;
template<typename T>inline void rd(T&x){
    x=0;int f=1;char ch=getchar();
    while(ch<'0' ||ch>'9'){ if(ch=='-')f=-1; ch=getchar(); }
    while(ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
    x*=f;
}
typedef long long ll;
typedef vector<pair<int, int> >vii;
typedef double db;
const int M = 2e6+7;
const int N = 26; 
const int MOD =51123987;
#define ls o*2
#define rs o*2+1
#define pb push_back
struct Pam {
	vii next[M];
	// int next[maxn][26];
	int fail[M];
	int len[M];
	int num[M];
	int cnt[M];
	int S[M];
	int last, n, p;
 
	int newNode(int l) {
		// memset(next[p], 0, sizeof(next[p]));
		next[p].clear();
		len[p] = l;
		return p++;
	}
 
	void init() {
		n = last = p = 0;
		newNode(0);
		newNode(-1);
		S[n] = -1;
		fail[0] = 1;
	}
 
	int getFail(int x) {
		while(S[n - len[x] - 1] != S[n]) {
			x = fail[x];
		}
		return x;
	}
 
	int find(int u, int c) {
		vii & x = next[u];
		int sz = x.size();
		for(int i = 0; i < sz; ++i) {
			if(x[i].first == c) return x[i].second;
		}
		return 0;
	}
 
	void add(int c) {
		S[++n] = c;
		int cur = getFail(last);
		int x = find(cur, c);
		if(!x) {
			int now = newNode(len[cur] + 2);
			x = now;
			fail[now] = find(getFail(fail[cur]), c);
			next[cur].push_back(make_pair(c, now));
			num[now] = num[fail[now]] + 1;
		}
		last = x;
		cnt[last]++;
	}
	void Count(){
        for(int i=p-1;i>=0;--i)cnt[fail[i]]+=cnt[i];
    }
} pam;
//p.p-2本质不同回文串的个数 
char s[M];
int hw[M];
int main()//不相交的回文串比较好算，用总数减去不相交即相交回文串个数 
{
	int len;
	scanf("%d%s",&len,s);
	pam.init();
	for(int i=0;i<len;i++)
	{
		pam.add(s[i]);
		hw[i]=pam.num[pam.last];
		int z=0;
		if(i-1>=0)z=hw[i-1];
		hw[i]=(hw[i]+z)%MOD;
	}
	pam.Count();
	ll ans=0;
	for(int i=2;i<pam.p;i++)
		ans=(ans+pam.cnt[i])%MOD;
	//cout<<pam.cnt[i]<<" ";//总回文串数 
	ans=ans*(ans-1)/2%MOD;
	reverse(s,s+len);
	pam.init();
	for(int i=0;i<len;i++)
	{	
		pam.add(s[i]);
		ll tmp,z=0;
		if(len-i-2>=0)
		z=hw[len-i-2];
		tmp=z*pam.num[pam.last]%MOD; 
		ans=(ans+MOD-tmp)%MOD;
	}
	printf("%lld\n",ans);
  	return 0;
}