后缀自动机重启之后缀自动机归来

1.endpos

对于字符串$s$，$s$的子串$t$，定义$endpos(t)$为一个集合，这个集合的元素为$t$每次在$s$中出现的结束位置。
例：$s=abcaabc$, e n d p o s ( a ) = { 1 , 4 , 5 } , e n d p o s ( b c ) = { 3 , 7 } endpos(a)=\{1,4,5\},endpos(bc)=\{3,7\} endpos(a)={1,4,5},endpos(bc)={3,7}。
特别的，定义 e n d p o s ( 空 串 ) = { 0 , 1 , 2 , . . . , n − 1 } endpos(空串)=\{0,1,2,...,n-1\} endpos(空串)={0,1,2,...,n−1}。

2.等价类

如果$s$的两个字符串$t1,t2$满足$endpos(t1)=endpos(t2)$，则称$t1,t2$属于同一个等价类。
设$s$的字串$t_1,t_2,...,t_{m-1},t_m(\forall i\ne j,t_i\ne t_j)$属于同一个等价类，且没有其它$s$的子串也属于这个等价类，则将字符串集$t_1,t_2,...,t_{m-1},t_m$称为$s$的一个等价类。后缀自动机中的每个结点都是一个等价类。
后缀自动机的结点数$=$等价类的数量（包括空串），将后缀自动机中结点$i$表示的等价类称为等价类$i$。
特别的，结点$1$表示空串，并称为结点$1$为起始节点。

3.引理

定义$suf(x,y)$表示串$x$是串$y$的后缀。
引理：对于$s$的字串$t_1,t_2,t_3$,满足$|t_1|<|t_2|<|t_3|$且$suf(t_1,t_2),suf(t_2,t_3）,endpos(t_1)=endpos(t_3)$，则$endpos(t_2)=endpos(t_1)=endpos(t_3)$。
证:显然$endpos(t_3)\subset endpos(t_2)\subset endpos(t_1)$，故若$endpos(t_1)=endpos(t_3)$必有结论成立。
推论：若$t_1,t_2,...,t_{m-1},t_m$为$s$的一个等价类，且$|t_1|<|t_2|<...<|t_{m-1}|<|t_m|$），则必有$|t_i|-|t_{i-1}|=1(1<i\le m),suf(t_{i-1},t_i)$。

4.link

设等价类$i$的字串集为$t_1,t_2,...,t_m$，设字串$t$包含于等价类$j$中，且满足$|t|=|t_1|-1,suf(t,t_1)$，则定义$link(i)=j$。
性质：当$link(i)=j$，有$endpos(t_1)⊊endpos(t)$，所有的$link$构成一颗以起始节点$1$为根的树。

5.nex

引理：设$t_1,t_2,...,t_m$为一个等价类，则对于任意的字符$c$，字串集$t_1+c,t_2+c,...,t_m+c$仍然属于同一个等价类（这个等价类可以是一个已经存在的等价类，也可以是一个新的等价类）。

若等价类$i$中的所有字符后面都接上一个字符$c$后，新的字串集属于等价类$j$，则称$nex(i,c)=j$（若$j$不存在，则$nex(i,c)=null$）。
$nex$称为后缀自动机的转移函数(从某个等价类转移到另一个等价类)。

6.len

定义$len(i)$表示等价类$i$的字串集中长度最大的字符串的长度。其最小长度为$len(link(i))+1$。
性质：$len(i)-len(link(i))$为等价类$i$中的字符串个数。

7.后缀自动机的构建

令字符串$s=s_1{s}_2{s}_3...s_n$，令$pre(x)=s_1{s}_2...s_x(1\le x\le n)$，令$pre(0)$表示空串，现要构建$s$的后缀自动机。
采用归纳法构造：
1.$pre(0)$为空串，其后缀自动机只有一个起始结点（等价类），其$nex=null,len=0,link=0$。
2.假设当前已经构建了$pre(i-1)$的后缀自动机，现构建$pre(i)$的后缀自动机，这等价于在子串$s_i,s_{i-1}{s}_i,s_{i-2}{s}_{i-1}{s}_i,...$的$endpos$中添加一个$i$，并且必然会添加一个新的结点（因为 e n d p o s ( p r e ( i ) ) = { i } endpos(pre(i))=\{i\} endpos(pre(i))={i}）。

引理：设构建$pre(i)$的后缀自动机时时新建的结点为$x$，则结点$x,link(x),link(link(x)),...,1$的字串集的并集为$pre(i)$的后缀集。

令$last$表示构建$pre(i-1)$时新建的结点，令$now$表示构建$pre(i)$时新建的结点，由$len(now)=len(last)+1$，再分以下情况考虑：
$nex(last)(s_i)=null$：在$pre(i-1)$所属的等价类后面新加字符$s_i$，新的子串集属于等价类$now$，令$nex(last)(s_i)=now$即可。由引理，继续递归检查$last$到$1$的$link$路径上的结点。

在$last$的$link$路径上存在结点$p$满足$nex(p)(s_i)=q\ne null$，且$p$的任意前驱结点$x$都满足$nex(x)(s_i)=null$：等价类$p$的字串集加上字符$s_i$后属于等价类$q$，但是等价类$q$中为$pre(i)$后缀的字串集新加入值$i$的$endpos$，又分两种情况讨论：
当$len(q)=len(p)+1$，则$q$的字符集都为$pre(i)$的后缀，又由$link$的定义，令$link(now)=q$即可。
当$len(q)>len(p)+1$，则$q$中长度大于$len(p)+1$的不是$pre(i)$的后缀，因此把长度$\le len(p)+1$这部分分离出来，构成一个新的结点$nq$，不难证明$link(q)=link(now)=nq,nex(nq)=nex(s)$$,link(nq)=p$。而对于$p$到$1$的$link$路径上的其它结点$op$，若$nex(op)(s_i)=q$，则令$nex(op)(s_i)=nq$，否则其一定属于$link(s)$路径上，也属于新构建后$link(now)$的路径上（显然$link(s)$之后的字串集都为$pre(i)$的后缀）。

#include<bits/stdc++.h>
using namespace std;
const int N=2e6+5;
int last=1,tot=1,fa[N],sum[N],endpos[N],len[N],t[N][26];
char s[N];
void add(int x)
{
    int p=last,n=++tot;
    last=n;endpos[n]=1;len[n]=len[p]+1;
    while(p&&!t[p][x]) t[p][x]=n,p=fa[p];
    if(!p) fa[n]=1;
    else
    {
        int q=t[p][x];
        if(len[p]+1==len[q]) fa[n]=q;
        else
        {
            int nq=++tot;
            for(int i=0;i<26;i++)
                t[nq][i]=t[q][i];
            fa[nq]=fa[q];fa[q]=nq;
            fa[n]=nq;len[nq]=len[p]+1;
            while(p&&t[p][x]==q) t[p][x]=nq,p=fa[p];
        }
    }
}
queue<int>q;
int main()
{
    scanf("%s",s+1);
    for(int i=1;s[i];i++)
        add(s[i]-'a');
    for(int i=2;i<=tot;i++)
        sum[fa[i]]++;
    for(int i=2;i<=tot;i++)
        if(!sum[i]) q.push(i);
    while(!q.empty())
    {
        int u=q.front();q.pop();
        endpos[fa[u]]+=endpos[u];
        sum[fa[u]]--;
        if(!sum[fa[u]]) q.push(fa[u]);
    }
    int ans=0;
    for(int i=2;i<=tot;i++)
        if(endpos[i]!=1)
        ans=max(ans,endpos[i]*len[i]);
    printf("%d\n",ans);
}

8.广义后缀自动机

当构造多串的后缀自动机时，只要注意，上述是用$pre(i-1)$的自动机构造$pre(i)$的自动机，令$pre(i-1)+c_1=pre(i)$，如果再构建一个$pre(i-1)+c_2=p$的自动机，那么只需要把$pre(i-1)+c_1$和$pre(i-1)+c_2$看成不同等价类即可（虽然它们的$endpos$相同，但是其字符串不相等）。
简单来说，只要对等价类定义做点小小的修改：$t_1,t_2$属于同一等价类，必须满足$suf(t_1,t_2)||suf(t_2,t_1)$，且$endpos(t_1)=endpos(t_2)$。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+5;
string s[N];
int n,k;
ll ans[N];
int las=1,tot=1,sa[N],f[N],vis[N],num[N],sum[N],len[N],son[N][26],fa[N];
int work(int p,int c)
{
    int nq=++tot,q=son[p][c];
    len[nq]=len[p]+1;
    fa[nq]=fa[q];fa[q]=nq;
    memcpy(son[nq],son[q],sizeof(son[q]));
    while(p&&son[p][c]==q) son[p][c]=nq,p=fa[p];
    return nq;
}
int ins(int p,int c)
{
    if(son[p][c])
    {
        int q=son[p][c];
        if(len[q]==len[p]+1) return q;
        else return work(p,c);
    }
    else
    {
        int np=++tot;len[np]=len[p]+1;
        while(p&&!son[p][c]) son[p][c]=np,p=fa[p];
        if(!p) fa[np]=1;
        else
        {
            int q=son[p][c];
            if(len[q]==len[p]+1) fa[np]=q;
            else fa[np]=work(p,c);
        }
        return np;
    }
}
void solve()
{
    int u;
    for(int i=1;i<=n;i++)
    {
        u=1;
        for(int j=0;j<s[i].size();j++)
        {
            u=son[u][s[i][j]-'a'];int p=u;
            while(p&&vis[p]!=i) sum[p]++,vis[p]=i,p=fa[p];
        }
    }
    for(int i=1;i<=tot;i++) num[len[i]]++;
    for(int i=1;i<=tot;i++) num[i]+=num[i-1];
    for(int i=tot;i>=1;i--) sa[num[len[i]]--]=i;
    for(int i=2;i<=tot;i++)
        u=sa[i],f[u]=f[fa[u]]+(sum[u]>=k?len[u]-len[fa[u]]:0);
    for(int i=1;i<=n;i++)
    {
        u=1;
        for(int j=0;j<s[i].size();j++)
            u=son[u][s[i][j]-'a'],ans[i]+=f[u];
    }
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>s[i];
        las=1;
        for(int j=0;j<s[i].size();j++)
            las=ins(las,s[i][j]-'a');
    }
    solve();
    for(int i=1;i<=n;i++)
        printf(i==n?"%lld\n":"%lld ",ans[i]);
}