bzoj-1030 文本生成器

题意：

给出一个n个单词的字典，单词长度<=100；

求长度为m的随机字符串中有多少个串至少包括一个单词；

n<=60，m<=100；

题解：

好久没有写AC自动机啦，如今还记得模板真是难得；

然而这似乎是Trie图？总之写出来不差几句话；

首先至少包括一个单词这个条件不太好弄；

那就转化一下，求不含单词的字符串个数；

然后就想办法处理这个；

定义状态，f[i][j]表示长度为i的字符串，最后几位的状态在Trie图上的结点编号；

那么转移显然了：

最开始f[0][1]是1；

从一个点到下一个点累加f；

如果目前结点代表的单词在字典中，f值为0；

有个小坑，就是可能有单词包括另一个单词，这时也要将大单词里的小单词标上标记；

就是在找fail的时候顺便标一下，具体可以看代码；

不过代码很丑= =，我反正是写不出来比较短的AC自动机。。。

代码：

#include<queue>#include<stdio.h>#include<string.h>#include<algorithm>#define N 128#define M 6000#define mod 10007#define pr pair<int,int>using namespace std;int next[M][26],fail[M],root=1,tot=1;int f[N][M];char str[N];bool is[M];int pow(int x,int y){int ret=1;while(y){if(y&1)ret=ret*x%mod;x=x*x%mod;y>>=1;}return ret;}void Insert(char *s){int p=root,index;while(*s!='\0'){index=*s-'A';if(!next[p][index])next[p][index]=++tot;p=next[p][index];s++;}is[p]=1;}void Build(){queue<int>q;int x,p,i,temp;q.push(root);while(!q.empty()){x=q.front(),q.pop();for(i=0;i<26;i++){p=next[x][i];temp=fail[x];while(temp){if(next[temp][i]){if(p&&is[next[temp][i]])is[p]=1;(p?fail[p]:next[x][i])=next[temp][i];break;}temp=fail[temp];}if(!temp)(p?fail[p]:next[x][i])=root;if(p)q.push(p);}}}int main(){int n,m,i,j,k,index,ans;scanf("%d%d",&n,&m);for(i=1;i<=n;i++){scanf("%s",str);Insert(str);}Build();f[0][1]=1;for(i=1;i<=m;i++){for(j=1;j<=tot;j++){for(k=0;k<26;k++){index=next[j][k];if(is[index])continue;f[i][index]=(f[i][index]+f[i-1][j])%mod;}}}for(i=1,ans=0;i<=tot;i++){ans+=f[m][i];ans%=mod;}printf("%d\n",((pow(26,m)-ans)%mod+mod)%mod);return 0;}