Solution -「SDOI 2017」「洛谷 P3706」硬币游戏

$\mathscr{Description}$

  Link.

  给定 $n$ 个长度为 $m$ 且两两不同的字符串 $S_{1..n}$, 字符集 $|\Sigma|=2$. 各位置独立在 $\Sigma$ 中均匀随机, 生成一个足够长的字符串 $T$, 对于每个 $S_i$, 求 $S_i$ 在 $T$ 中第一次出现位置最靠前的概率.

  $n,m\le300$.

$\mathscr{Solution}$

  感觉一直知道 PGF 但基本没有上手用过 qwq.

  比较套路啦, 令 $F_i(x)$ 表示 $S_i$ 出现位置最靠前时, 其概率关于 $|T|$ 的 PGF; $G(x)$ 表示没有任何 $S$ 出现时, 其概率关于 $|T|$ 的 PGF. 我们的目标是求所有 $F_i(1)$.

  "$T_i$ 被放出来的概率" = "$|T|=i-1$ 时没有结束的概率":

$$G(x)+\sum_iF_i(x)=1+xG(x).$$

  "从一个未结束状态向后再恰好延伸出一个 $S_i$ 的概率" = "在放的过程中所有戛然而止 (某个 $S_j$ 出现了) 的时候再向后强行补满未出现字符的概率":

$$2^{-m}x^mG(x)=\sum_{j=1}^n\sum_{k=1}^m[S_i[:k]=S_j[m-k+1:]]2^{k-m}x^{m-k}F_j(x).$$

  一起令 $x=1$:

$$\begin{cases} G(1)+\sum_iF_i(1)=1+G(1),\\ 2^{-m}G(1)=\sum_j\sum_k[S_i[:k]=S_j[m-k+1:]]2^{k-m}F_j(1)&\forall i. \end{cases}$$

直接对总共 $n+1$ 个变量消元即可. 复杂度 $\mathcal O(n^2(n+m))$.

$\mathscr{Code}$

/*+Rainybunny+*/

#include <bits/stdc++.h>

#define rep(i, l, r) for (int i = l, rep##i = r; i <= rep##i; ++i)
#define per(i, r, l) for (int i = r, per##i = l; i >= per##i; --i)

typedef unsigned long long ULL;

const int MAXN = 300;
const ULL BASE = 127;
int n, m;
ULL hpw[MAXN + 5], hsh[MAXN + 5][MAXN + 5];
double mat[MAXN + 5][MAXN + 5], pwr[MAXN + 5], ans[MAXN + 5];

inline void init() {
    hpw[0] = 1;
    rep (i, 1, m) hpw[i] = hpw[i - 1] * BASE;
    pwr[0] = 1;
    rep (i, 1, m) pwr[i] = 0.5 * pwr[i - 1];
}

inline ULL calc(const int i, const int l, const int r) {
    return hsh[i][r] - hpw[r - l + 1] * hsh[i][l - 1];
}

inline void gauss() {
    rep (i, 0, n) {
        int p = i;
        rep (j, i + 1, n) if (fabs(mat[p][i]) < fabs(mat[j][i])) p = j;
        if (i != p) std::swap(mat[p], mat[i]);
        rep (j, i + 1, n) {
            double t = mat[j][i] / mat[i][i];
            rep (k, i, n + 1) mat[j][k] -= t * mat[i][k];
        }
    }
    per (i, n, 0) {
        ans[i] = mat[i][n + 1] / mat[i][i];
        rep (j, 0, i - 1) mat[j][n + 1] -= ans[i] * mat[j][i];
    }
}

int main() {
    scanf("%d %d", &n, &m), init();
    rep (i, 1, n) {
        static char str[MAXN + 5];
        scanf("%s", str + 1);
        rep (j, 1, m) hsh[i][j] = hsh[i][j - 1] * BASE + str[j];
    }

    rep (i, 1, n) {
        mat[i][0] = -pwr[m];
        rep (j, 1, n) {
            rep (k, 1, m) if (calc(i, 1, k) == calc(j, m - k + 1, m)) {
                // printf("%d %d %d\n", i, j, k);
                mat[i][j] += pwr[m - k];
            }
        }
    }
    rep (i, 1, n + 1) mat[0][i] = 1.;

    gauss();
    rep (i, 1, n) printf("%.12f\n", ans[i]);
    return 0;
}