【BZOJ】4861: [Beijing2017]魔法咒语-AC自动机+DP+矩乘

传送门：bzoj4861

---

题解

首先把忌讳词建成AC自动机，所有串的$end$结点和可以通过跳$fail$链到一个$end$结点的点都不能走到($getfail$时预处理即可)。

当$L\le 100$时，考虑在AC自动机上DP：
DP算是比较套路的了。
设$f[i][j]$表示已经选择了$i$个字符，走到了自动机上$j$结点的方案数，有
$f[j+s{z}_k][trans[j][k]]+=f[i][j]$
其中$k$枚举的是基本词汇，$s{z}_k$是串长，$trans[j][k]$表示在$j$结点时再接上串$k$后会走到的结点。
设初始值$f[0][0]=1$后$O(n^3)DP$即可。

当$L>100$时，$DP$的复杂度显然不可接受，但考虑到$len\le 2$，
也就是一个最多关乎$i-1,i-2$两项的线性递推式，直接矩阵快速幂即可。
复杂度$O(n^{3}logL)$

---

代码

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7,N=110;

int n,m,L,ans,cnt,trans[N][N];
int sz[N],len,mx,ed[N];
char s[N][N],t[N];

inline void ad(int &x,int y){x+=y;if(x>=mod) x-=mod;}

struct AC{
    queue<int>Q;
	int ch[N][26],f[N];
	
	inline void ins(char *s)
	{
		int i,alp,u=0;len=strlen(s);
		for(i=0;i<len;++i){
			alp=s[i]-'a';
			if(!ch[u][alp]) ch[u][alp]=++cnt;
			u=ch[u][alp];
		}
		ed[u]=1;
	}
	
	void getfail()
	{
		int i,x,y,z,u;
		for(i=0;i<26;++i){
			x=ch[0][i];if(x) Q.push(x);
		}
		for(;!Q.empty();){
			x=Q.front();Q.pop();y=f[x];
			for(i=0;i<26;++i){
				z=ch[x][i];u=ch[y][i];
				if(!z) {ch[x][i]=u;continue;}
				f[z]=u;ed[z]|=ed[u];Q.push(z);
			}
		}
	}
	
	inline void getrans(int id,char *s)
	{
		int i,j,u;len=strlen(s);
		mx=max(mx,len);sz[id]=len;
		for(i=0;i<len;++i) s[i]-='a';
		for(i=0;i<=cnt;++i) if(!ed[i]){
			u=i;
			for(j=0;j<len;++j){
			  u=ch[u][(int)s[j]];
			  if(ed[u]) {u=-1;break;}	
			}
			trans[i][id]=u;
		}
	}
}ac;

namespace pA{
    int dp[N][N];
    
	void solve(){
		int i,j,k,res;
		dp[0][0]=1;
	    for(i=0;i<L;++i)
	      for(j=0;j<=cnt;++j)
	       if(dp[i][j]){
	  	       res=dp[i][j];
	  	       for(k=1;k<=n;++k) 
				 if((~trans[j][k]) && i+sz[k]<=L)
	  	 	       ad(dp[i+sz[k]][trans[j][k]],res);
	       }
	    for(i=0;i<=cnt;++i) ad(ans,dp[L][i]);
	    printf("%d",ans);
	}
}

namespace pB{
	int lim;
	
	struct Mat{
		int v[N<<1][N<<1];
		inline void itia(){for(int i=0;i<=lim;++i) v[i][i]=1;}
	}g,e,re;
	
	inline void Mul(Mat &x,Mat y)
	{
		int i,j,k,res;
		for(i=0;i<=lim;++i)
		 for(j=0;j<=lim;++j){
		 	res=0;
		 	for(k=0;k<=lim;++k) 
			  ad(res,1ll*x.v[i][k]*y.v[k][j]%mod);
			re.v[i][j]=res;
		 }
		for(i=0;i<=lim;++i)
		 for(j=0;j<=lim;++j)
		  x.v[i][j]=re.v[i][j];
	}
	
	void maker()
	{
		int i,j;
		if(mx==1){
		   lim=cnt;
		   for(i=0;i<=cnt;++i) 
		    if(!ed[i])
		      for(j=1;j<=n;++j) 
			    if(~trans[i][j])
		   	     g.v[i][trans[i][j]]++;
		}else{
		   lim=(cnt<<1)+1;
		   for(i=0;i<=cnt;++i)
		    for(j=1;j<=n;++j)
		     if(~trans[i][j]){
		     	if(sz[j]==1)
		     	 g.v[i][trans[i][j]]++;
		     	else
		     	 g.v[i+cnt+1][trans[i][j]]++;
		     }
		   for(i=0;i<=cnt;++i) g.v[i][i+cnt+1]++;
		}
	}
	
	void solve()
	{
		int i,j;
		maker();e.itia();
		for(;L;L>>=1,Mul(g,g))
		 if(L&1) Mul(e,g);
		for(i=0;i<=cnt;++i){
			ad(ans,e.v[0][i]);
		}
		printf("%d",ans);
	}
}

int main(){
	int i;scanf("%d%d%d",&n,&m,&L);
	for(i=1;i<=n;++i) scanf("%s",s[i]);
	for(i=1;i<=m;++i){scanf("%s",t);ac.ins(t);}
	ac.getfail();memset(trans,0xff,sizeof(trans));
	for(i=1;i<=n;++i) ac.getrans(i,s[i]);
	if(L<=100) pA::solve();
	else pB::solve();
	return 0;
}