bzoj4820概率dp(kmp +高消)

xiaoxiao2022-05-11  84

题面

题解

思想很重要: 最暴力的是p[i]表示AC自动机上每轮经过每个点的概率和(期望次数) 所有终止节点的概率之和=1。注意对于1号节点(AC自动机的根),概率要+1,因为一开始从它出发

然后我们发现对于所有非终止节点,我们统计每轮经过它们的概率和是没有任何意义的,我们可以把这些点放在一起考虑

所以就有了题解中的做法 只考虑在任一非终止节点后+串A,提前出现串B的概率

注意:概率高消eps越小越好?甚至设成0,然后系数选绝对值最大的优先,要不然精度会很爆炸

#include <bits/stdc++.h> #define rep(i,a,b) for(int i=a;i<=b;++i) #define per(i,a,b) for(int i=a;i>=b;--i) #define repd(i,a,b) for(int i=a;i>=b;--i) #define rvc(i,S) for(int i=0;i<(int)S.size();++i) #define fore(i,x) for(int i = head[x] ; i ; i = e[i].next) #define mp make_pair #define pb push_back #define fi first #define se second #define debug(...) fprintf(stderr,__VA_ARGS__) #define lowbit(x) (x&(-x)) using namespace std; #define maxn 1020 #define N 90020 #define eps 0 #define long double double double mat[maxn][maxn],f[maxn]; int n,T,l,a[maxn][maxn],s[maxn * 2],fail[maxn * 2];; char ch[maxn]; double pow2[maxn]; void getfail(int a[],int n){ fail[1] = 0; rep(i,2,n){ int p = fail[i - 1]; while ( p && a[i] != a[p + 1] ) p = fail[p]; fail[i] = a[i] == a[p + 1] ? p + 1 : 0; } } double kmp(int a[],int b[]){ int len = l * 2; rep(i,1,l) s[i] = a[i]; rep(i,1,l) s[i + l] = b[i]; getfail(s,len); double res = 0; int x = fail[len]; while ( x ){ res += pow2[l - x]; x = fail[x]; } /* while ( x ){ if ( fail[x] <= l && fail[x] ) res += pow2[l - fail[x]]; x = fail[x]; }*/ return res; } void build(){ pow2[0] = 1; rep(i,1,l) pow2[i] = pow2[i - 1] / 2; rep(i,1,n){ rep(j,1,n){ mat[i][j] = kmp(a[i],a[j]); // else mat[i][j] = 1; } mat[i][n + 1] = -pow2[l]; } rep(i,1,n + 2) mat[n + 1][i] = 1; mat[n + 1][n + 1] = 0; } void gauss(){ n++; /*rep(i,1,n){ rep(j,1,n + 1){ cout<<mat[i][j]<<" "; } cout<<endl; } */ rep(i,1,n){ int x = 0; rep(j,i,n){ if ( fabs(mat[j][i]) > eps ){ x = j; break; } } if ( fabs(mat[x][i]) < eps ) continue; if ( x != i ) rep(j,1,n + 1) swap(mat[x][j],mat[i][j]); rep(j,1,n){ if ( i != j && fabs(mat[j][i]) > eps ){ double tmp = mat[j][i] / mat[i][i]; rep(k,1,n + 1){ mat[j][k] -= tmp * mat[i][k]; } } } } /* rep(i,1,n){ rep(j,1,n + 1){ cout<<mat[i][j]<<" "; } cout<<endl; }*/ repd(i,n,1){ // if ( fabs(mat[i][i]) < eps ) continue; rep(j,i + 1,n) mat[i][n + 1] -= mat[i][j] * f[j]; f[i] = mat[i][n + 1] / mat[i][i]; } // rep(i,1,n) cout<<f[i]<<" "; // cout<<endl; } int main(){ // freopen("input.txt","r",stdin); scanf("%d %d",&n,&l); rep(i,1,n){ scanf("%s",ch + 1); rep(j,1,l){ if ( ch[j] == 'H' ) a[i][j] = 0; else a[i][j] = 1; } } build(); gauss(); rep(i,1,n - 1){ printf("%.10lf\n",(double)f[i]); } return 0; }
转载请注明原文地址: https://www.6miu.com/read-4881537.html

最新回复(0)