1030: [JSOI2007]文本生成器 ac自动机+dp

时间:2021-03-29 22:32:29

https://www.lydsy.com/JudgeOnline/problem.php?id=1030

求长度为m不包含n个子串的种数,

跑完ac自动机之后没办法跑矩阵快速幂,因为状态数比较大(6000),所以dp转移,dp[i][j]表示前i个跑到j状态的不包含子串的情况数

//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pi acos(-1.0)
#define ll long long
#define vi vector<int>
//#define mod 1000000007
#define C 0.5772156649
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define pil pair<int,ll>
#define pli pair<ll,int>
#define pii pair<int,int>
#define cd complex<double>
#define ull unsigned long long
#define base 1000000000000000000
#define fio ios::sync_with_stdio(false);cin.tie(0) using namespace std; const double g=10.0,eps=1e-;
const int N=+,maxn=+,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f; ll n,m,dp[][N];
ll quick(ll a,ll b,ll c)
{
ll ans=;
while(b)
{
if(b&)ans=ans*a%c;
a=a*a%c;
b>>=;
}
return ans;
}
char s[N];
struct ACM{
int root,tot;
int Next[N][],fail[N],End[N];
int newnode()
{
memset(Next[tot],-,sizeof Next[tot]);
End[tot]=;
return tot++;
}
ACM()
{
tot=;
root=newnode();
}
void ins()
{
int now=root,len=strlen(s);
for(int i=;i<len;i++)
{
if(Next[now][s[i]-'A']==-)
Next[now][s[i]-'A']=newnode();
now=Next[now][s[i]-'A'];
}
End[now]=;
}
void build()
{
queue<int>q;
fail[root]=root;
for(int i=;i<;i++)
{
if(Next[root][i]==-)Next[root][i]=root;
else
{
fail[Next[root][i]]=root;
q.push(Next[root][i]);
}
}
while(!q.empty())
{
int now=q.front();
q.pop();
if(End[fail[now]])End[now]=;
for(int i=;i<;i++)
{
if(Next[now][i]==-)Next[now][i]=Next[fail[now]][i];
else
{
fail[Next[now][i]]=Next[fail[now]][i];
q.push(Next[now][i]);
}
}
}
}
void solve()
{
dp[][]=;
for(int i=;i<=m;i++)
{
for(int j=;j<tot;j++)
{
if(!End[j]&&dp[i-][j])
{
for(int k=;k<;k++)
{
int now=j;
while(Next[now][k]==-)now=fail[now];
dp[i][Next[now][k]]=(dp[i][Next[now][k]]+dp[i-][j])%;
}
}
}
}
ll ans=quick(,m,);
for(int i=;i<tot;i++)
if(!End[i])
ans=((ans-dp[m][i])%+)%;
printf("%lld\n",ans);
}
}ac;
int main()
{
scanf("%lld%lld",&n,&m);
for(int i=;i<n;i++)
{
scanf("%s",s);
ac.ins();
}
ac.build();
ac.solve();
return ;
}
/*********************** ***********************/