题意
对于字符串 S S S 的每个前缀,求其所以回文子串的前缀个数和,本质相同的前缀不重复计数。 N ≤ 3 × 1 0 5 N\leq3\times 10^5 N≤3×105。
题解
建议先阅读 【乱搞】某 SCOI 模拟赛 T3 串(string)【PAM 乱搞】
在 Anti-Hack 的基础上,把 vector
维护每个儿子改成 set
维护每个儿子在 SA 里的
r
a
n
k
rank
rank。于是每次出现新节点只需要判断与它 rk
最接近的两个兄弟与它的 LCP,复杂度至多是
O
(
log
n
)
O(\log n)
O(logn),实际上很难卡满。
在出题人的数据上跑得比复杂度最高的那个暴力还慢。
代码:
#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;
}