【SDOI2017】硬币游戏【KMP】【概率期望】【高斯消元】
題意:給 nnn 個(gè)長度為 mmm 的 01 串,一個(gè) 01 串初始為空,不斷隨機(jī)一個(gè)字符加在后面,當(dāng)出現(xiàn)給定的 nnn 個(gè)串中的一個(gè)時(shí)停止。分別求在 nnn 個(gè)串處停止的概率。
考場(chǎng)思路歷程:
顯然建出 AC 自動(dòng)機(jī)(flag++)
然后發(fā)現(xiàn)變成了 CF113D Museum,有個(gè) O((nm)3)O((nm)^3)O((nm)3) 的做法
考慮方程有什么特殊性質(zhì)
也就是說是 Linear Creature 的二叉樹版。
然后發(fā)現(xiàn)要求任意點(diǎn)出發(fā)所有點(diǎn)的出現(xiàn)次數(shù)期望,不弱于原問題,無法推廣。
但是是豎著放的,沒時(shí)間了沒仔細(xì)想。
正解:
既然方程的特殊性質(zhì)不好優(yōu)化,考慮直接想辦法求出 nnn 個(gè)答案的方程。
設(shè) PiP_{i}Pi? 為第 iii 個(gè)串的期望出現(xiàn)次數(shù),即概率,即答案。特別地,記 P0P_0P0? 為所有未終止?fàn)顟B(tài)(即 AC 自動(dòng)機(jī)上無標(biāo)記的點(diǎn))的期望長度,即期望出現(xiàn)次數(shù)之和。
考慮對(duì) PiP_iPi? 建立方程。
無視掉走到就終止的條件,考慮所有未終止?fàn)顟B(tài),往后面按 SiS_iSi? 走 mmm 步,得到了一個(gè)狀態(tài)的可重集,注意有些狀態(tài)可能不合法但也算在內(nèi)(不過都在 AC 自動(dòng)機(jī)上)。
從這個(gè)過程來考慮,因?yàn)橄拗屏撕竺?mmm 步的走法,這些狀態(tài)的期望出現(xiàn)次數(shù)之和就是 12mP0\frac{1}{2^m}P_02m1?P0?。
如果過程中沒有到過終止?fàn)顟B(tài),那么就是 PiP_iPi?。否則我們枚舉第一個(gè)終止?fàn)顟B(tài)以及是走了幾步達(dá)到的,設(shè)達(dá)到的串為 SjS_jSj?(注意 i,ji,ji,j 可以相等),是走了 kkk 步后達(dá)到的,因?yàn)樗写L相等,所以此時(shí)一定是合法狀態(tài)。然后就在 PjP_jPj? 的基礎(chǔ)上繼續(xù)走 m?km-km?k 步。
得到方程:
12mP0=Pi+∑j=1n∑k=1m?1[Si,1~k=Sj,m?k+1~m]Pj2m?k\frac{1}{2^m}P_0=P_i+\sum_{j=1}^n\sum_{k=1}^{m-1}[S_{i,1\sim k}=S_{j,m-k+1\sim m}]\frac{P_j}{2^{m-k}}2m1?P0?=Pi?+j=1∑n?k=1∑m?1?[Si,1~k?=Sj,m?k+1~m?]2m?kPj??
然后對(duì)每對(duì) i,ji,ji,j 跑個(gè) kmp 就可以了。
什么?終止?fàn)顟B(tài)的長度小于 m?km-km?k 怎么辦?反正只有有限個(gè),忽略好了。
然后再加一個(gè) ∑i=1nPi=1\sum _{i=1}^ nP_i=1∑i=1n?Pi?=1 的方程來得到數(shù)值,一共有 n+1n+1n+1 個(gè)方程和 n+1n+1n+1 個(gè)未知數(shù),高斯消元解出來就可以了。
#include <iostream> #include <cstdio> #include <cstring> #include <cctype> #include <cmath> #define MAXN 305 using namespace std; char s[MAXN][MAXN]; int n,m; double pw[MAXN],a[MAXN][MAXN]; int nxt[MAXN]; void gauss(int n) {for (int i=0;i<n;i++){int pos=i;for (int j=i+1;j<=n;j++) if (fabs(a[j][i])>fabs(a[pos][i])) pos=j;if (pos>i) swap(a[i],a[pos]);for (int j=0;j<=n;j++)if (i!=j){double t=a[j][i]/a[i][i];for (int k=i;k<=n;k++) a[j][k]-=t*a[i][k];}} } int main() {scanf("%d%d",&n,&m);for (int i=1;i<=n;i++) scanf("%s",s[i]+1);pw[0]=1;for (int i=1;i<=m;i++) pw[i]=pw[i-1]/2;for (int i=1;i<=n;i++){a[i][0]=-pw[m];a[i][i]+=1;int pos=0;for (int j=2;j<=m;j++){while (pos&&s[i][j]!=s[i][pos+1]) pos=nxt[pos];if (s[i][j]==s[i][pos+1]) nxt[j]=++pos;else nxt[j]=0;}for (int j=1;j<=n;j++){pos=0;for (int k=1;k<=m;k++){while (pos&&s[j][k]!=s[i][pos+1]) pos=nxt[pos];if (s[j][k]==s[i][pos+1]) ++pos;}if (i==j) pos=nxt[pos];for (;pos;a[i][j]+=pw[m-pos],pos=nxt[pos]);}a[n+1][i]=1;}a[n+1][n+1]=1;gauss(n+1);for (int i=1;i<=n;i++) printf("%.10f\n",a[i][n+1]/a[i][i]);return 0; }總結(jié)
以上是生活随笔為你收集整理的【SDOI2017】硬币游戏【KMP】【概率期望】【高斯消元】的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【CF113D】Museum【概率期望】
- 下一篇: 【ZJOI2019】线段树【线段树上dp