CF700E Cool Slogans 后缀自动机 + right集合线段树合并 + 树形DP

Posted 2022-03-19 guangheli

又是一道 SAM 神题.

Code: 

#include<bits/stdc++.h>
#define maxn 400002 
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;   
namespace tr 

    #define lson t[x].l 
    #define rson t[x].r 
    #define mid ((l+r)>>1) 
    int cnt; 
    struct Node  int l,r; t[maxn*20]; 
    void modify(int &x,int l,int r,int k)
    
        if(!x) x=++cnt; 
        if(l==r) return; 
        if(k<=mid) modify(lson,l,mid,k); 
        else modify(rson,mid+1,r,k); 
    
    int merge(int u,int v) 
    
        if(!u||!v) return u+v; 
        int x=++cnt; 
        lson=merge(t[u].l,t[v].l); 
        rson=merge(t[u].r,t[v].r);    
        return x;             
    
    int query(int x,int l,int r,int L,int R)
    
        if(!x) return 0; 
        if(l>=L&&r<=R) return 1; 
        int tmp=0; 
        if(L<=mid) tmp+=query(lson,l,mid,L,R); 
        if(R>mid) tmp+=query(rson,mid+1,r,L,R); 
        return tmp; 
    
; 
namespace SAM 
 
    char str[maxn]; 
    int last,tot,n;  
    int trans[maxn][27],f[maxn],len[maxn],C[maxn],rk[maxn],top[maxn],dp[maxn],pos[maxn],rt[maxn];  
    void init()  last=tot=1; 
    void extend(int c,int i) 
      
        int np=++tot,p=last; 
        len[np]=len[p]+1,last=np; 
        while(p&&!trans[p][c]) trans[p][c]=np,p=f[p]; 
        if(!p) f[np]=1; 
        else 
        
            int q=trans[p][c]; 
            if(len[q]==len[p]+1) f[np]=q; 
            else 
            
                int nq=++tot; 
                len[nq]=len[p]+1; 
                pos[nq]=pos[q];  
                memcpy(trans[nq],trans[q],sizeof(trans[q])); 
                f[nq]=f[q], f[np]=f[q]=nq; 
                while(p&&trans[p][c]==q) trans[p][c]=nq,p=f[p]; 
            
        
        pos[np]=i; 
        tr::modify(rt[last],1,n,i);   
    
    void prepare() 
    
        int i,j; 
        init(); 
        scanf("%d%s",&n,str+1);  
        for(i=1;i<=n;++i) extend(str[i]-‘a‘,i);  
        for(i=1;i<=tot;++i) ++C[len[i]]; 
        for(i=1;i<=tot;++i) C[i]+=C[i-1]; 
        for(i=1;i<=tot;++i) rk[C[len[i]]--]=i;    
        for(i=tot;i>1;--i) 
        
            int u=rk[i]; 
            rt[f[u]]=tr::merge(rt[f[u]],rt[u]);  
                             
    
    void calc() 
    
        int i,j,ans=1; 
        for(i=2;i<=tot;++i) 
        
            int u=rk[i],ff=f[rk[i]];  
            if(ff==1)  top[u]=u,dp[u]=1; continue;   
            if(tr::query(rt[top[ff]],1,n,pos[u]-len[u]+len[top[ff]],pos[u]-1))
                top[u]=u,dp[u]=dp[top[ff]]+1; 
            else 
                top[u]=top[ff];       
            ans=max(ans,dp[u]);                 
        
        printf("%d\n",ans);  
    
; 
int main()

    //msetIO("input"); 
    SAM::prepare();                  
    SAM::calc(); 
    return 0;