给出N个由大写字母组成的字符串,再给出一个长度M
求有多少个由大写字母组成的长度为M的字符串满足“至少包含这N个字符串中的一个”
A包含B指的是,B是A的子串
答案对10007取模
N<=60,每个字符串的长度<=100,M<=100
解析:这道题正着想显然是有点难,那我们就倒过来,用所有可能的文章总数减去不包含给定字符串的数量。
设F[i][j]表示匹配前i个字母之后走到了AC自动机的j号节点的方案数
则有对于任意的j的儿子节点x
若x不是危险节点,则F[i+1][x]+=F[i][j]
最后把所有不是危险节点的F[M][x]求和即是不包含这N个字符串的串的数量
用26^M减去即可
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<iostream>
#include<iomanip>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
const int mod=10007;
struct Trie
{
Trie *son[26],*fail;
bool match;
int num;
Trie()
{
memset(son,0,sizeof(son));
fail=NULL;
match=false;
}
}mempool[50000],*root=&mempool[0];
int top=0;
void my_insert(char *s)
{
Trie *o=root;
while(*s)
{
if(!o->son[*s-'A']) o->son[*s-'A']=&mempool[++top];
o=o->son[*s-'A'];
s++;
}
o->match=true;
}
void bfs()
{
static Trie* dui[50000];
int top=1,my_final=1;
for(int i=0;i<26;i++)
{
if(root->son[i])
{
root->son[i]->fail=root;
dui[my_final++]=root->son[i];
}
else root->son[i]=root;
}
while(top<my_final)
{
Trie *o=dui[top];
for(int i=0;i<26;i++)
{
if(o->son[i])
{
o->son[i]->fail=o->fail->son[i];
dui[my_final++]=o->son[i];
}
else o->son[i]=o->fail->son[i];
}
top++;
Trie *t=o->fail;
while(t!=root && !t->match) t=t->fail;
o->fail=t;
if(t->match) o->match=true;
}
}
int f[50000][101];
char s[10000];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%s",s);
my_insert(s);
}
bfs();
for(int i=0;i<=top;i++) mempool[i].num=i;
f[0][0]=1;
int ans=0;
for(int k=0;k<m;k++)
for(int i=0;i<=top;i++)
{
if(mempool[i].match) continue;
for(int j=0;j<26;j++)
{
if(!mempool[i].son[j]->match)
f[mempool[i].son[j]->num][k+1]=(f[mempool[i].son[j]->num][k+1]+f[i][k])%mod;
}
}
for(int i=0;i<=top;i++) ans=(ans+f[i][m])%mod;
int t=m;
int mid=26;
int trueans=1;
while(t)
{
if(t&1) trueans=(trueans*mid)%mod;
mid=(mid*mid)%mod;
t>>=1;
}
cout<<(trueans-ans+mod)%mod;
return 0;
}