[Jsoi2009]有趣的游戏【概率DP】【AC自动机+高斯消元】

题目链接：https://www.lydsy.com/JudgeOnline/problem.php?id=1444

首先构建AC自动机。

然后问题就变成了经典问题：

给一个有环图，0号点为起点，每条边有一定的概率去走，有若干个终点，到达终点后必须停止，问经过某一点的概率是多少。

然后就可以用高斯消元解决了。

 

（我怀疑这题数据水）

#include<bits/stdc++.h>
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
using namespace std;
const int MAX = 550;
const int N = 500;
const int SIGMA_SIZE = 26;

int n,l,m;
double P[50];
char s[20];
int t[20];
double a[500][500];
double ans[500];


struct ACAM {
    int ch[MAX][SIGMA_SIZE];
    int val[MAX], last[MAX], f[MAX], sz;
    int ANS;
    void init() {
        sz = 1;                         //节点个数
        memset(ch, 0, sizeof(ch));      //路径
        memset(val, 0, sizeof(val));    //结束标记
        memset(f, 0, sizeof(f));        //失配指针
        memset(last, 0, sizeof(last));  //当前节点的上一个模式串
    }
    void add(int u) {
        while (u) {
            ANS += val[u];
            u = last[u];
        }
    }
    void creat(char *s,int x) {
        int u = 0, len = strlen(s);
        for (int i = 0; i < len; i++) {
            int c = s[i]-'A';
            if (!ch[u][c]) ch[u][c] = sz++;
            u = ch[u][c];
        }
        val[u]++;
        t[x] = u;
    }
    void get_fail() {           //找失配指针
        queue<int> q;
        for (int i = 0; i < SIGMA_SIZE; i++)
            if (ch[0][i]) q.push(ch[0][i]);
        while (!q.empty()) {
            int r = q.front();
            q.pop();
            for (int c = 0; c < SIGMA_SIZE; c++) {
                int u = ch[r][c];
                if (!u) {
                    ch[r][c] = ch[f[r]][c];
                    continue;
                }
                q.push(u);
                int v = f[r];
                while (v && ch[v][c] == 0) v = f[v];
                f[u] = ch[v][c];
                last[u] = val[f[u]] ? f[u] : last[f[u]];
            }
        }
    }
}AC;

bool gauss(int n) {
    rep(i, 0, n-1) {          //逐列消除
        int p = i;
        rep(j, i+1, n-1)          //找到该列最大系数所在的行p,减少误差(也可以找第一个非0系数,但误差会大一些)
            if(fabs(a[p][i])<fabs(a[j][i]))
                p = j;

        swap(a[i],a[p]);        //否则把p交换上来,用其进行消元

        double div = a[i][i];
        rep(j, i, n) a[i][j] /= div;  //把p的第i列系数置1
        rep(j, 0, n-1)  {                 //开始消其他行
            if(i==j) continue;
            div = a[j][i];
            rep(k, i, n)
                a[j][k] -= a[i][k]*div;
        }
    }
    rep(i, 0, n-1) ans[i] = a[i][n+1];
    return 1;
}
int main() {
   // freopen("a.txt","r",stdin);
    ios::sync_with_stdio(0);
    AC.init();
    cin>>n>>l>>m;
    rep(i, 0, m-1) {
        double x,y;
        cin>>x>>y;
        P[i] = x/y;
    }
    rep(i, 1, n) {
        cin>>s;
        AC.creat(s,i);
    }
    AC.get_fail();
    int sz = AC.sz;
    a[0][0] = 1;
    a[0][sz] = 1;
    rep(u, 0, sz-1) a[u][u] = 1;
    rep(u, 0, sz-1) {
        if(AC.val[u]) continue;
        rep(j, 0, m-1) {
            int v = AC.ch[u][j];
            a[v][u] -= P[j];
        }
    }
    gauss(AC.sz);
    rep(i, 1, n) printf("%.2f\n",a[t[i]][sz]);
    return 0;
}