spoj8222 NSUBSTR - Substrings（SAM+dp）

我们首先建出后缀自动机，每个节点的|Right|就是这个节点代表的串的出现次数。可以基数排序做出拓扑序之后倒着更新sz。考虑dp[i]表示长度为i的串最多的出现次数，显然有dp[i]不升。于是我们每次更新一个节点时，其实只要更新dp[mx[i]]即可，最后再倒着更新一遍dp[i]=max(dp[i],dp[i+1])即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 250010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
char s[N];
int n=0,m,par[N<<1],mx[N<<1],sz[N<<1],last=0,rt=0,son[N<<1][26];
int a[N<<1],cnt[N],dp[N];
inline void ins(int ch){
    int p=last,np=++n;last=np;mx[np]=mx[p]+1;sz[np]=1;
    for(;p&&!son[p][ch];p=par[p]) son[p][ch]=np;
    if(!p){par[np]=rt;return;}
    int q=son[p][ch];
    if(mx[q]==mx[p]+1){par[np]=q;return;}
    int nq=++n;mx[nq]=mx[p]+1;memcpy(son[nq],son[q],sizeof(son[q]));
    par[nq]=par[q];par[q]=par[np]=nq;
    for(;p&&son[p][ch]==q;p=par[p]) son[p][ch]=nq;
}
int main(){
//  freopen("a.in","r",stdin);
    scanf("%s",s+1);m=strlen(s+1);last=rt=++n;
    for(int i=1;i<=m;++i) ins(s[i]-'a');
    for(int i=1;i<=n;++i) cnt[mx[i]]++;
    for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
    for(int i=n;i>=1;--i) a[cnt[mx[i]]--]=i;
    for(int i=n;i>=1;--i){
        int x=a[i];sz[par[x]]+=sz[x];
        dp[mx[x]]=max(dp[mx[x]],sz[x]);
    }for(int i=m-1;i>=1;--i) dp[i]=max(dp[i],dp[i+1]);
    for(int i=1;i<=m;++i) printf("%d\n",dp[i]);
    return 0;
}