某 SCOI 模拟赛 T3 串（string）【PAM】

题意

对于字符串 $S$ 的每个前缀，求其所以回文子串的前缀个数和，本质相同的前缀不重复计数。$N\le 3\times 10^5$。

题解

建议先阅读 【乱搞】某 SCOI 模拟赛 T3 串（string）【PAM 乱搞】

在 Anti-Hack 的基础上，把 vector 维护每个儿子改成 set 维护每个儿子在 SA 里的 $rank$。于是每次出现新节点只需要判断与它 rk 最接近的两个兄弟与它的 LCP，复杂度至多是 $O(\log n)$，实际上很难卡满。

在出题人的数据上跑得比复杂度最高的那个暴力还慢。

代码：

#include<bits/stdc++.h>
using namespace std;
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans*f;
}
const int N=3e5+10;
char s[N];
int n;

namespace SA{

int sa[N],rk[N],tp[N],h[N];
int c[N];
void rsort(int m){
	for(int i=1;i<=m;i++)c[i]=0;
	for(int i=1;i<=n;i++)c[rk[tp[i]]]++;
	for(int i=1;i<=m;i++)c[i]+=c[i-1];
	for(int i=n;i>=1;i--)sa[c[rk[tp[i]]]--]=tp[i];
}
bool cmp(int x,int y,int k){
	return tp[x]==tp[y]&&tp[x+k]==tp[y+k];
}
void get_sa(){
	for(int i=1;i<=n;i++)rk[i]=s[i],tp[i]=i;
	int m='z';
	rsort(m);
	for(int p=0,k=1;p<n;k<<=1,m=p){
		p=0;
		for(int i=n-k+1;i<=n;i++)tp[++p]=i;
		for(int i=1;i<=n;i++)
			if(sa[i]>k)tp[++p]=sa[i]-k;
		rsort(m);
		memcpy(tp,rk,sizeof(tp));
		rk[sa[1]]=p=1;
		for(int i=2;i<=n;i++)
			rk[sa[i]]=cmp(sa[i],sa[i-1],k)?p:++p;
	}
	//for(int i=1;i<=n;i++)cerr<<sa[i]<<" ";cerr<<endl;
	//for(int i=1;i<=n;i++)cerr<<rk[i]<<" ";cerr<<endl;
	int t=0;
	for(int i=1;i<=n;i++){
		if(t)--t;
		while(s[sa[rk[i]-1]+t]==s[i+t])++t;
		h[rk[i]]=t;
	}
	//for(int i=1;i<=n;i++)cerr<<h[i]<<" ";cerr<<endl;
}
const int L=20;
int st[L][N],l2[N];
void init_st(){
	l2[0]=-1;for(int i=1;i<=n;i++)st[0][i]=h[i+1],l2[i]=l2[i>>1]+1;
	for(int i=1;i<L;i++){
		for(int j=1;j<=n-(1<<i-1);j++){
			st[i][j]=min(st[i-1][j],st[i-1][j+(1<<i-1)]);
		}
	}
}
int get_min(int l,int r){
	if(l>r)swap(l,r);
	if(l==r)return n-sa[l]+1;
	int t=l2[r-l];
	return min(st[t][l],st[t][r-(1<<t)]);
}

}

namespace PAM{

int ch[N][26],fail[N],len[N],cnt,lst,po[N];
set<int> fch[N];
string str[N];
long long ans=0;
void init(){
	cnt=2;
	fail[0]=1;len[0]=0;fch[1].insert(0);
	fail[1]=1;len[1]=-1;
}
long long qaq=0;
void extend(char c,int pos){
	int cur=lst;
	while(s[pos-len[cur]-1]!=c)cur=fail[cur];
	if(!ch[cur][c-'a']){
		lst=cnt++;
		len[lst]=len[cur]+2;
		po[lst]=pos-len[lst]+1;
		fail[lst]=fail[cur];
		while(s[pos-len[fail[lst]]-1]!=c)fail[lst]=fail[fail[lst]];
		fail[lst]=ch[fail[lst]][c-'a'];
		//fch[fail[lst]].push_back(lst);
		ch[cur][c-'a']=lst;
		if(0);else{
			int fa=fail[lst];
			auto it=fch[fa].insert(SA::rk[po[lst]]).first;
			int l=len[fa];
			if(it!=fch[fa].begin()){
				auto itt=it;--itt;
				l=max(l,SA::get_min(*itt,*it));
			}
			auto ittt=it;++ittt;
			if(ittt!=fch[fa].end()){
				l=max(l,SA::get_min(*ittt,*it));
			}
			ans+=len[lst]-l;
		}
	}
	lst=ch[cur][c-'a'];
}
void print(){
	for(int i=0;i<cnt;i++){
		for(int j=0;j<26;j++){
			if(ch[i][j])cerr<<i<<" "<<ch[i][j]<<" "<<char('a'+j)<<endl;
		}
		cerr<<"fail "<<i<<" "<<fail[i]<<endl;
	}
}

}//namespace PAM

int main(){
	//freopen("TEST\\string.in","r",stdin);
	//freopen("string.out","w",stdout);
	n=getint();
	scanf("%s",s+1);
	PAM::init();
	SA::get_sa();
	SA::init_st();
	for(int i=1;i<=n;i++){
		PAM::extend(s[i],i);
		printf("%d\n",PAM::ans);
	}
	//print();
	//cerr<<PAM::qaq;
	return 0;
}