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]);
}