zoj 3535 Gao the String II(AC自动机+DP)

题意：给出两个字符串集合A，B，现在每次可以在集合A中挑一个字符串接在当前组成的串S后(可以重叠),对于每个长度不超过L的S串，S中每出现一个B中的串，那么这个串的价值加一，求串的最大的价值。

思路：把B插入Trie建立AC自动机。然后进行dp，dp[i][j][k]表示长度为i的串，最后接的一个串是j，此时状态为k时串的最大价值，然后对于每个状态，枚举一下能接下去的串就行了，注意接上去的串要至少让长度增加一，另外如果覆盖了上一个串，实际上这是没有意义的，所以不用考虑。本来以为复杂度会很高，但实际上没有那么糟糕。还有，给出的B的串有可能重复，这里要注意一下。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=555;
const int csize=26;
int ch[maxn][csize],lastv[maxn],flag[maxn];
int next[maxn],cnt[maxn],size;
char str[maxn],astr[55][15];
int m,n,L,dp[55][55][maxn],conect[55][55][15],lena[55];
void Init()
{
    memset(ch[0],0,sizeof(ch[0]));
    memset(cnt,0,sizeof(cnt));
    memset(lastv,0,sizeof(lastv));
    memset(flag,0,sizeof(flag));
    memset(next,0,sizeof(next));
    size=0;
}
void Insert(const char *s)
{
    int u=0;
    for(int i=0;s[i];++i)
    {
        int c=s[i]-'a';
        if(!ch[u][c])
        {
            ch[u][c]=++size;
            memset(ch[size],0,sizeof(ch[size]));
        }
        u=ch[u][c];
    }
    flag[u]++;
}
void build()
{
    queue<int>q;
    for(int i=0;i<csize;++i)
        if(ch[0][i]) q.push(ch[0][i]);
    int r,u;
    while(!q.empty())
    {
        r=q.front();q.pop();
        for(int c=0;c<csize;++c)
        {
            u=ch[r][c];
            if(!u) {ch[r][c]=ch[next[r]][c];continue;}
            q.push(u);
            int j=next[r];
            while(j&&!ch[j][c]) j=next[j];
            next[u]=ch[j][c];
            lastv[u]=flag[next[u]]?next[u]:lastv[next[u]];
        }
    }
    for(int i=1;i<=size;++i)
    {
        u=i;
        if(flag[u]||lastv[u])
        {
            u=flag[u]?u:lastv[u];
            while(u)
            {
                cnt[i]+=flag[u];
                u=lastv[u];
            }
        }
    }
}
bool check(int x,int y,int z)
{
    int lenx=lena[x];
    if(z>lenx) return false;
    for(int i=0;i<z;++i)
    {
        if(astr[x][lenx-z+i]!=astr[y][i]) return false;
    }
    return true;
}
int solve()
{
    lena[0]=0;
    memset(conect,0,sizeof(conect));
    for(int i=1;i<=m;++i)
    {
        conect[0][i][0]=1,conect[0][i][1]=0;
        lena[i]=strlen(astr[i]);
    }
    for(int i=1;i<=m;++i)
        for(int j=1;j<=m;++j)
        {
            conect[i][j][0]=1;
            conect[i][j][1]=0;
            for(int k=1;k<lena[j];++k)
            {
                if(check(i,j,k)) {conect[i][j][0]++;conect[i][j][conect[i][j][0]]=k;}
            }
        }
    memset(dp,0xff,sizeof(dp));
    dp[0][0][0]=0;
    int ans=0,u,c,tmp;
    for(int i=0;i<=L;++i)
        for(int j=0;j<=m;++j)
            for(int k=0;k<=size;++k)
            {
                if(j==0&&i!=0) continue;
                if(dp[i][j][k]==-1) continue;
                if(dp[i][j][k]>ans) ans=dp[i][j][k];
                for(int x=1;x<=m;++x)
                    for(int y=1;y<=conect[j][x][0];++y)
                    {
                        u=k;tmp=0;
                        for(int z=conect[j][x][y];astr[x][z];++z)
                        {
                            c=astr[x][z]-'a';
                            u=ch[u][c];
                            tmp+=cnt[u];
                        }
                        int pos=i+lena[x]-conect[j][x][y];
                        if(pos>L) continue;
                        if(dp[pos][x][u]==-1) dp[pos][x][u]=dp[i][j][k]+tmp;
                        else dp[pos][x][u]=max(dp[pos][x][u],dp[i][j][k]+tmp);
                    }
            }
    return ans;
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    while(~scanf("%d%d%d",&m,&n,&L))
    {
        Init();
        astr[0][0]='\0';
        for(int i=1;i<=m;++i)
            scanf("%s",astr[i]);
        for(int i=0;i<n;++i)
        {
            scanf("%s",str);
            Insert(str);
        }
        build();
        int ans=solve();
        printf("%d\n",ans);
    }
    return 0;
}