【Hihocoder1413】Rikka with String（后缀自动机）

【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;
}

转载于:https://www.cnblogs.com/cjyyb/p/10187446.html