【BZOJ】4861: [Beijing2017]魔法咒语 AC自动机+DP+矩阵快速幂

【题意】给定n个原串和m个禁忌串，要求用原串集合能拼出的不含禁忌串且长度为L的串的数量。(60%)n,m<=50,L<=100。(40%)原串长度为1或2，L<=10^18。

【算法】AC自动机+DP+矩阵快速幂

【题解】其实题意的数据范围不太清晰，反正开200个点就足够了。

因为要匹配禁忌串，所以对禁忌串集合建立AC自动机，标记禁忌串结尾节点，以及下传到所有能fail到的点（这些点访问到都相当于匹配了禁忌串）。

令f[i][j]表示匹配到节点i，长度为j的串的数量，先预处理a[i][j]表示节点 i 匹配第 j 个原串到达的节点编号，那么就有：

f [ a[i][j] ] [ L+size[j] ] += f [ i ] [ L ]

以上就是60%数据的做法，对于40%的数据使用矩阵快速幂。

假设原串长度均为1，那么DP的转移如下：

$$f[i][L]=\sum_{j}f[j][L-1]\ \ ,\ \ j \rightarrow i$$

这很容易用一个长度为第一维大小（AC自动机节点数）的矩阵维护转移，第L个列向量就是f[i][L]。

如果原串长度有2，那么再记录L-1即可。

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=,MOD=1e9+;
int n,m,a[maxn][],ch[maxn][],val[maxn],size[maxn],sz=,fail[maxn];
ll L;
char s[][maxn],S[maxn];
queue<int>Q;
void insert(char *s){
    int n=strlen(s),u=;
    for(int i=;i<n;i++){
        int c=s[i]-'a';
        if(!ch[u][c])ch[u][c]=++sz;
        u=ch[u][c];
    }
    val[u]++;
}
void AC_build(){
    for(int c=;c<;c++)if(ch[][c])Q.push(ch[][c]);
    while(!Q.empty()){
        int u=Q.front();Q.pop();
        for(int c=;c<;c++)if(ch[u][c]){
            fail[ch[u][c]]=ch[fail[u]][c];
            Q.push(ch[u][c]);
            val[ch[u][c]]|=val[fail[ch[u][c]]];//
        }
        else ch[u][c]=ch[fail[u]][c];
    }
}
int M(int x){return x>=MOD?x-MOD:x;}
namespace Task1{
    int f[maxn][];
    void solve(){
        f[][]=;
        for(int l=;l<L;l++){//
            for(int i=;i<=sz;i++)if(f[i][l]){
                for(int j=;j<=n;j++)if(~a[i][j]&&l+size[j]<=L){
                    f[a[i][j]][l+size[j]]=M(f[a[i][j]][l+size[j]]+f[i][l]);
                }
            }
        }
        int ans=;
        for(int i=;i<=sz;i++)if(f[i][L]&&!val[i])ans=M(ans+f[i][L]);
        printf("%d",ans);
    }
}
namespace Task2{
    const int maxn=;
    int N,A[maxn*][maxn*],ANS[maxn*][maxn*],c[maxn*][maxn*];
    void mul(int a[maxn*][maxn*],int b[maxn*][maxn*]){
        for(int i=;i<=N;i++){
            for(int j=;j<=N;j++){
                c[i][j]=;
                for(int k=;k<=N;k++)c[i][j]=M(c[i][j]+1ll*a[i][k]*b[k][j]%MOD);
            }
        }
        for(int i=;i<=N;i++)for(int j=;j<=N;j++)b[i][j]=c[i][j];
    }
    void solve(){
        N=sz*+;
        for(int i=;i<=sz;i++){
            for(int j=;j<=n;j++)if(~a[i][j]){
                if(size[j]==)A[a[i][j]*][i*]++;
                else A[a[i][j]*][i*+]++;
            }
            A[i*+][i*]=;
        }
        ANS[][]=;
        while(L){
            if(L&)mul(A,ANS);
            mul(A,A);
            L>>=;
        }
        int ans=;
        for(int i=;i<=sz;i++)if(!val[i])ans=M(ans+ANS[i*][]);
        printf("%d",ans);
    }
}
int main(){
    scanf("%d%d%lld",&n,&m,&L);
    for(int i=;i<=n;i++)scanf("%s",s[i]);
    for(int i=;i<=m;i++){
        scanf("%s",S);
        insert(S);
    }
    AC_build();
    memset(a,-,sizeof(a));
    for(int k=;k<=n;k++){
        size[k]=strlen(s[k]);
        for(int i=;i<=sz;i++){
            int u=i;
            for(int j=;j<size[k];j++)if(!val[u])u=ch[u][s[k][j]-'a'];else break;
            if(!val[u])a[i][k]=u;
        }
    }
    if(L<=)Task1::solve();else
    Task2::solve();
    return ;
}