bzoj1444 有趣的游戏(AC自动机+矩阵乘法)

多串比较，又和概率相关，一定是AC自动机维护概率dp

但会发现dp转移顺序一言难尽。。。

于是考虑把方程放到矩阵中高斯消元

但你又会发现，有个偷懒的做法，就是把得到的矩阵自乘，乘上很多次便可无限接近答案

数据小，不卡精，水果。。。

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,len,cnt,tot;
char s[15];
int pos[15];
bool ed[105];
double p[15];
struct Trie{
    int son[10];
    int fail;
}tr[105];
struct node{
    double f[105][105];
}ans;
void build(char a[],int mrk){
    int now=0;
    for(int i=1;i<=len;i++){
        int k=a[i]-'A';
        if(!tr[now].son[k])tr[now].son[k]=++tot;
        now=tr[now].son[k];
    }
    ed[now]=true;
    pos[mrk]=now;
}
void getfail(){
    queue<int>que;
    for(int i=0;i<10;i++){
        if(tr[0].son[i])que.push(tr[0].son[i]);
    }
    while(!que.empty()){
        int u=que.front();
        que.pop();
        for(int i=0;i<10;i++){
            if(tr[u].son[i]){
                tr[tr[u].son[i]].fail=tr[tr[u].fail].son[i];
                que.push(tr[u].son[i]);
            }
            else tr[u].son[i]=tr[tr[u].fail].son[i];
        }
    }
}
void get_squ(){
    for(int i=0;i<=tot;i++){
        if(ed[i]){
            ans.f[i][i]=1;
            continue;
        }
        for(int j=0;j<10;j++){
            ans.f[i][tr[i].son[j]]+=p[j];
        }
    }
}
void mul(){
    node c;
    for(int i=0;i<=tot;i++){
        for(int j=0;j<=tot;j++){
            c.f[i][j]=0.00;
        }
    }
    for(int i=0;i<=tot;i++){
        for(int j=0;j<=tot;j++){
            for(int k=0;k<=tot;k++){
                c.f[i][j]+=ans.f[i][k]*ans.f[k][j];
            }
        }
    }
    for(int i=0;i<=tot;i++){
        for(int j=0;j<=tot;j++){
            ans.f[i][j]=c.f[i][j];
        }
    }
}
int main(){
    scanf("%d%d%d",&n,&len,&m);
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        p[i-1]=(double)x/(double)y;
    }
    for(int i=1;i<=n;i++){
        scanf("%s",s+1);
        build(s,i);
    }
    getfail();get_squ();
    for(int i=1;i<=50;i++)mul();
    for(int i=1;i<=n;i++){
        printf("%.2lf
",ans.f[0][pos[i]]);
    }
    return 0;
}