tyvj 1519 博彩游戏 AC自动机+DP

背景 Background
Bob最近迷上了一个博彩游戏……

描述 Description
这个游戏的规则是这样的：
每花一块钱可以得到一个随机数R，花上N块钱就可以得到一个随机序列；
有M个序列，如果某个序列是产生的随机序列的子串，那么就中奖了，否则不中。
Bob会告诉你这M个序列，和身上有的钱的总数N，当然还有R的范围。
请你告诉Bob中奖的概率有多少？

输入格式 InputFormat
第一行三个用空格隔开的数N、M和R的范围R。
其中1<=R<=9，0

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int N=4e5+5;
int val[N],fail[N],trie[N][30],tot=1,r,n,m;
char ch[30];
int q[N];
ll ans1,ans2,f[200][N];
inline void insert()
{
    int len=strlen(ch),x=1;
    fo(i,0,len-1)
    {
        int c=ch[i]-'0';
        if (trie[x][c])x=trie[x][c];
        else x=trie[x][c]=++tot;
    }
    val[x]=1;
}
inline void getfail()
{
    int t=0,w=1;
    q[0]=1;fail[1]=0;
    while (t<w)
    {
        int x=q[t++];
        fo(i,1,r)
        if (trie[x][i])
        {
            int y=trie[x][i],j=fail[x];
            while(j&&!trie[j][i])j=fail[j];
            //j=trie[j][i];
            fail[trie[x][i]]=trie[j][i];
            if (val[trie[j][i]])val[trie[x][i]]=1;
            q[w++]=trie[x][i];
        }
    }
}
inline void dp(int x)
{
    fo(i,1,tot)
    if (!val[i]&&f[x-1][i])
        fo(j,1,r)
        {
            int k=i;
            while (!trie[k][j])k=fail[k];
            f[x][trie[k][j]]+=f[x-1][i];
        }
}
int main()
{
    scanf("%d%d%d",&n,&m,&r);
    fo(i,1,r)trie[0][i]=1;
    fo(i,1,m)
    {
        scanf("%s",ch);
        insert();
    }
    getfail();
    f[0][1]=1;
    fo(i,1,n)dp(i);
    ans2=1;
    fo(i,1,n)ans2=ans2*r;
    fo(i,1,tot)if (!val[i])ans1+=f[n][i];
    printf("%.5lf",double(ans2-ans1)/ans2);
    return 0;
}