【Hihocoder1413】Rikka with String(后缀自动机)
题面
Hihocoder
给定一个小写字母串,回答分别把每个位置上的字符替换为'#'后的本质不同的子串数。
题解
首先横跨'#'左右的串一定恰好只会出现一次,所以直接统计答案。
那么剩下的部分就是左右的本质不同的子串数。
我们把答案拆成三个部分,先是左侧本质不同子串的个数,再是右侧本质不同子串的个数。最后再减去既在左侧出现过,又在右侧出现过的串的个数。
左右两个直接用(SAM)算就好了。
在两侧同时出现过的,我们维护每一个节点最小的(endpos)和最大的(endpos),显然'#'出现在两串之间的时候就会在左右同时被计算。
每个(endpos)表示的都是一段区间,这里就是做多个区间的区间加,所以二阶差分计算。
具体的东西看代码。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long
#define MAX 300300
int n;char ch[MAX];
struct Node
{
int son[26];
int ff,len;
}t[MAX<<1];
int last=1,tot=1;
ll ans[MAX],c1[MAX],c2[MAX],cnt=0;
int l[MAX<<1],r[MAX<<1];
void extend(int c,int id)
{
int p=last,np=++tot;last=np;
t[np].len=t[p].len+1;
while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].ff;
if(!p)t[np].ff=1;
else
{
int q=t[p].son[c];
if(t[q].len==t[p].len+1)t[np].ff=q;
else
{
int nq=++tot;
t[nq]=t[q];t[nq].len=t[p].len+1;
t[q].ff=t[np].ff=nq;
while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].ff;
}
}
cnt+=t[np].len-t[t[np].ff].len;
l[np]=r[np]=id;
}
int a[MAX<<1],p[MAX<<1];
int main()
{
scanf("%d",&n);scanf("%s",ch+1);
memset(l,63,sizeof(l));memset(r,0,sizeof(r));
for(int i=1;i<=n;++i)extend(ch[i]-97,i),ans[i+1]+=cnt;
for(int i=1;i<=n;++i)ans[i]+=1ll*i*(n-i+1);
for(int i=1;i<=tot;++i)a[t[i].len]++;
for(int i=1;i<=n;++i)a[i]+=a[i-1];
for(int i=1;i<=tot;++i)p[a[t[i].len]--]=i;
for(int i=tot;i>1;--i)
{
int u=p[i];
l[t[u].ff]=min(l[t[u].ff],l[u]);
r[t[u].ff]=max(r[t[u].ff],r[u]);
int len=min(r[u]-l[u],t[u].len);
if(len<=t[t[u].ff].len)continue;
c1[l[u]+1]+=len-t[t[u].ff].len;
c2[r[u]-t[t[u].ff].len+1]+=1;
c2[r[u]-len+1]-=1;
}
for(int i=1;i<=n;++i)c2[i]+=c2[i-1];
for(int i=1;i<=n;++i)c1[i]+=c1[i-1]+c2[i];
for(int i=1;i<=n;++i)ans[i]-=c1[i];
memset(t,0,sizeof(t));last=tot=1;cnt=0;
for(int i=n;i;--i)extend(ch[i]-97,i),ans[i-1]+=cnt;
for(int i=1;i<=n;++i)printf("%lld ",ans[i]);
puts("");return 0;
}