2019徐州网络赛 G colorful string 回文自动机+前缀和

给出$n\le 3e5$的字符串，求出所有的回文子串的价值之和，价值的定义是回文串中不同字母的数量。
预处理$26$种不同字母的前缀和，每次查询暴力查询$26$种字母存在的个数。然后回文树上求出所有本质不同的回文串出现的次数，同时预处理出每个回文串的位置。最后直接遍历所有回文子串，答案就是$\sum cal(l[i],r[i])\ast cnt[i]$。
这里发现回文树也是可以方便处理回文串的本身的性质的，可以记录下串的末尾位置，转化为区间问题。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
const int N=3e5+5;

struct Palindromic_Tree {
	int nxt[N][26],fail[N],len[N];
	int num[N],cnt[N],S[N];
	int endpos[N];
	int lst,n,p;
	int newnode(int l) {
		for(int i=0;i<26;i++) nxt[p][i]=0;
		cnt[p]=num[p]=0;
		endpos[p]=0;
		len[p]=l;
		return p++;
	}
	void init() {
		p=0;
		newnode(0);
		newnode(-1);
		lst=n=0;
		S[0]=-1;
		fail[0]=1;
	}
	int getfail(int x) {
		while(S[n-len[x]-1]!=S[n]) x=fail[x];
		return x;
	}
	void add(int c,int i) {
		c-='a';
		S[++n]=c;
		int cur=getfail(lst);
		if(!nxt[cur][c]) {
			int now=newnode(len[cur]+2);
			fail[now]=nxt[getfail(fail[cur])][c];
			nxt[cur][c]=now;
			num[now]=num[fail[now]]+1;
		}
		lst=nxt[cur][c];
		endpos[lst]=i;
		cnt[lst]++;
	}
	void count() {
		for(int i=p-1;i>=0;i--) 
			cnt[fail[i]]+=cnt[i]; 
	}
}pam;
char str[N];
int sum[N][26];
int cal(int l,int r) {
	int ans=0;
	for(int i=0;i<26;i++) {
		if(sum[r][i]-sum[l-1][i]>0)
			ans++;
	}
	return ans;
}
int main() {
	pam.init();
	scanf("%s",str+1);
	int len=strlen(str+1);
	for(int i=1;i<=len;i++) {
		for(int j=0;j<26;j++) {
			if(str[i]-'a'==j) sum[i][j]=sum[i-1][j]+1;
			else sum[i][j]=sum[i-1][j];
		}
	} 
	for(int i=1;i<=len;i++) {
		pam.add(str[i],i);
	}
	pam.count(); 
	ll ans=0;
	for(int i=2;i<pam.p;i++) {
		int r=pam.endpos[i];
		int l=pam.endpos[i]-pam.len[i]+1;
		ans+=1LL*cal(l,r)*pam.cnt[i];
	}
	printf("%lld\n",ans); 
	return 0;
}