UVALive - 4811 Growing Strings (AC自动机+dp)

题目链接：UVALive - 4811 Growing Strings

题意：

给你n个字符串，问你最多能选出多少个字符串，使得s[i]是s[i+1]的子串。

题解：

先将所有的字符串插入AC自动机，将所有字符串按长度排序后，显然dp[i]=max{dp[j]}+1,其中s[j]是s[i]的子串。然后就完了。

#include<bits/stdc++.h>
#define mst(a,b) memset(a,b,sizeof(a))
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

const int AC_N=1e6+7,tyn=26;
int n,dp[AC_N],ed,ans;
char s[AC_N];
struct Str
{
    int l,len;
    bool operator<(const Str &B)const{return len<B.len;}
}str[AC_N];

struct AC_automation{
    int tr[AC_N][tyn],cnt[AC_N],Q[AC_N],fail[AC_N],tot;
    inline int getid(char x){return x-'a';}
    void nw(){cnt[++tot]=0,fail[tot]=0;mst(tr[tot],0);}
    void init(){tot=-1,fail[0]=-1,nw();}
    void insert(char *s,int len,int idx,int x=0){
        for(int i=0,w;i<len;x=tr[x][w],i++)
            if(!tr[x][w=getid(s[i])])nw(),tr[x][w]=tot;
        cnt[x]=idx;
    }
    void build(int head=1,int tail=0){
        for(int i=0;i<tyn;i++)if(tr[0][i])Q[++tail]=tr[0][i];
        while(head<=tail)for(int x=Q[head++],i=0;i<tyn;i++)
            if(tr[x][i])fail[tr[x][i]]=tr[fail[x]][i],Q[++tail]=tr[x][i];
            else tr[x][i]=tr[fail[x]][i];
    }
    int ask(char *s,int l,int &ans){
        for(int x=0,i=0,w,j;i<l;i++){
            x=tr[x][w=getid(s[i])];
            for(j=x;j;j=fail[j])if(cnt[j])ans=max(ans,dp[cnt[j]]);
        }
        return ans;
    }
}AC;

int main(){
    while(scanf("%d",&n),n)
    {
        ed=0,ans=1;
        F(i,1,n)
        {
            scanf("%s",s+ed);
            str[i]=Str{ed,(int)strlen(s+ed)};
            ed+=str[i].len;
        }
        sort(str+1,str+1+n);
        AC.init();
        F(i,1,n)AC.insert(s+str[i].l,str[i].len,i);
        AC.build(),dp[1]=1;
        F(i,2,n)
        {
            dp[i]=0,AC.ask(s+str[i].l,str[i].len,dp[i]);
            dp[i]++,ans=max(ans,dp[i]);
        }
        printf("%d\n",ans);
    }
    return 0;
}