BZOJ 1444: [Jsoi2009]有趣的游戏 [AC自动机 高斯消元]

1444: [Jsoi2009]有趣的游戏

题意：每种字母出现概率\(p_i\)，有一些长度len的字符串，求他们出现的概率

---

套路DP的话，\(f[i][j]\) i个字符走到节点j的概率，建出转移矩阵来矩乘几十次可以认为是无穷个字符，就得到概率了

但我们发现Trie图也是图啊，直接高斯消元就好了，\(f[i]\)表示走到节点i的期望次数

注意\(f[0]\)需要+1

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=105;
const double eps=1e-8;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}

int n, len, m, pos[N]; double p[N], x, y;
char s[N];
namespace ac{
    struct meow{int ch[11], fail, val;} t[N];
    int sz;
    void insert(char *s, int id) {
        int u=0;
        for(int i=1; i<=len; i++) {
            int c=s[i]-'A';
            if(!t[u].ch[c]) t[u].ch[c] = ++sz;
            u=t[u].ch[c]; 
        }
        t[u].val=1; 
        pos[id]=u;
    }

    int q[N], head, tail;
    void build() {
        head=tail=1;
        for(int i=0; i<m; i++) if(t[0].ch[i]) q[tail++] = t[0].ch[i];
        while(head!=tail) {
            int u=q[head++];
            t[u].val |= t[t[u].fail].val;
            for(int i=0; i<m; i++) {
                int &v = t[u].ch[i];
                if(!v) v = t[t[u].fail].ch[i];
                else t[v].fail = t[t[u].fail].ch[i], q[tail++]=v;
            }
        }
    }
}using ac::t; using ac::sz;

double a[N][N];
namespace eq{
    void build() { 
        a[0][sz+1] = 1;
        for(int i=0; i<=sz; i++) {  //printf("i %d\n",i);
            a[i][i]=1;
            if(!t[i].val) for(int j=0; j<m; j++) 
                a[ t[i].ch[j] ][i] -= p[j];// printf("ch %d %lf  %d\n",j,p[j],t[i].ch[j]);
        }
        //for(int i=0; i<=n; i++) for(int j=0; j<=n+1; j++) printf("%lf%c",a[i][j],j==n+1?'\n':' ');
    }

    void gauss(int n) {
        for(int i=0; i<=n; i++) {
            int r=i;
            for(int j=i; j<=n; j++) if(abs(a[j][i])>abs(a[r][i])) r=j;
            if(r!=i) for(int j=0; j<=n+1; j++) swap(a[r][j], a[i][j]);

            for(int k=i+1; k<=n; k++) {
                double t = a[k][i]/a[i][i];
                for(int j=i; j<=n+1; j++) a[k][j] -= t*a[i][j];
            }
        }
        for(int i=n; i>=0; i--) {
            for(int j=n; j>i; j--) a[i][n+1] -= a[i][j]*a[j][n+1];
            a[i][n+1] /= a[i][i];
        }
    }
}
int main() {
    freopen("in","r",stdin);
    n=read(); len=read(); m=read(); 
    int flag=0;
    for(int i=0; i<m; i++) x=read(), y=read(), p[i]=(double)x/y, flag |= p[i]>eps;
    if(!flag) {for(int i=1; i<=n; i++) puts("0.00"); return 0;}

    for(int i=1; i<=n; i++) scanf("%s",s+1), ac::insert(s, i);
    ac::build();
    eq::build(); eq::gauss(sz);
    //for(int i=1; i<=n; i++) printf("%d ",pos[i]); puts(" pos");
    for(int i=1; i<=n; i++) printf("%.2lf\n", a[pos[i]][sz+1]);
}