SPOJ8222 NSUBSTR - Substrings 后缀自动机_动态规划

Posted 2021-02-04 guangheli

讲起来不是特别好讲.
总之，如果 $dp[i+1]>=dp[i]$，故$dp[i]=max(dp[i],dp[i+1])$

 

Code:

#include <cstdio>
#include <algorithm>
#include <cstring>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 2000000
#define N 30
#define ll long long
using namespace std;
int last=1,tot=1,n;
int ch[maxn][N],cnt[maxn],f[maxn],dis[maxn],rk[maxn],dp[maxn]; 
ll C[maxn]; 
char str[maxn];
struct Suffix_Automaton{
    void ins(int c){
        int p=last,np=++tot; last=np; dis[np]=dis[p]+1;
        while(p&&!ch[p][c])ch[p][c]=np,p=f[p];
        if(!p) f[np]=1;
        else{
            int q=ch[p][c],nq; 
            if(dis[q]==dis[p]+1) f[np]=q;
            else{
                nq=++tot;
                dis[nq]=dis[p]+1;
                memcpy(ch[nq],ch[q],sizeof(ch[q]));
                f[nq]=f[q],f[q]=f[np]=nq;
                while(p&&ch[p][c]==q) ch[p][c]=nq,p=f[p];
            }
        }
        cnt[last]=1;
    }
}sam;
int main(){
    //setIO("input"); 
    scanf("%s",str),n=strlen(str);
    for(int i=0;i<n;++i) sam.ins(str[i]-‘a‘); 
    for(int i=1;i<=tot;++i) ++C[dis[i]];
    for(int i=1;i<=tot;++i) C[i]+=C[i-1]; 
    for(int i=1;i<=tot;++i) rk[C[dis[i]]--]=i;
    for(int i=tot;i>=1;--i) 
    {
        int p=rk[i];
        cnt[f[p]]+=cnt[p];
        dp[dis[p]]=max(dp[dis[p]],cnt[p]); 
    }
    for(int i=n-1;i>=1;--i) dp[i]=max(dp[i],dp[i+1]);
    for(int i=1;i<=n;++i) printf("%d
",dp[i]);
    return 0; 
}