[SDOI2017]硬币游戏

Solution

• 将所有可能的硬币序列分成 $3$ 类：
  $1.$终止态：生成这个序列后刚好结束游戏。
  $2.$非终止态：生成这个序列后还没结束游戏。
  $3.$非法态：这个序列还没生成完就结束游戏，即存在一个比它短的前缀是终止态。

• 记 $P_i$ 表示第 $i$ 个人获胜的概率，记 $P_0$ 表示生成一个非终止态序列的概率。记 $A$ 为任意一个非终止态序列，可以为空。

• $n+1$ 个变量需要 $n+1$ 个方程，先列一个 ${\sum }_{i=1}^n{P}_i=1$。

• 剩下的举个例子：
  $S_1=THT,S_2=TTH,S_3=HTT$
  使 $2$ 获胜的串一定是 $ATTH$，但 $ATTH$ 可能是非法态。

  $P(2$ 获胜$)=$ $P($生成以 $TTH$ 为后缀的终止态串$)$
  $=P($生成 $ATTH)-P($生成 $ATTH$，且 $AT$ 或 $ATT$ 为终止态的串，即 $ATTH$ 是非法态$)$

  $(1).AT$ 为终止态的条件是 $A$ 有后缀 $TH$ 或 $HT$，这样 $AT$ 会使 $1$ 或 $3$ 获胜，生成一个终止态 $AT$ 的概率就是 $P_1+P_3$，再生成一个 $TH$ 的概率是 $0.25$， 于是生成这种 $ATTH$ 的概率为 $0.25(P_1+P_3)$。

  $(2).ATT$ 为终止态的条件是 $A$ 有后缀 $H$，这样会使 $3$ 获胜，生成一个终止态 $ATT$ 的概率就是 $P_3$，于是生成这种 $ATTH$ 的概率为 $0.5P_3$。

  $(3).P($生成 $ATTH)=0.125P_0$。

  于是就有：$P_2=0.125P_0-0.25P_1-0.75P_3$。

• 类似地，可以列出 $n$ 个 $P_i=...$ 的方程。可以发现，如果 $S_i$ 有一个前缀和 $S_j$ 的一个后缀相等，则会对方程 $P_i=...$ 中 $P_j$ 的系数有贡献。

• 注意当$S_i$ 有一个长度小于本身的前缀后缀相同，则方程右边也会出现 $P_i$，例如 $P_1$，如果 $A$ 以 $TH$ 为后缀，那么 $ATHT$ 也是一个非法态。

Code

#include <bits/stdc++.h>

using namespace std;

#define ll long long

const ll z1 = 29, z2 = 31;
const int e = 605, h1 = 1e9 + 7, h2 = 1e9 + 9;
double p[e], b[e], a[e][e], ans[e];
int nxt[e], m, n, len, s1[e][e], s2[e][e], p1[e], p2[e];
char s[e][e], c[e];

inline void gauss()
{
   int i, j, k;
   for (i = 1; i < n; i++)
   {
       int x = i;
       for (j = i; j <= n; j++) if (fabs(a[j][i]) > fabs(a[x][i])) x = j;
       if (a[x][i] == 0) continue;
       if (x != i)
       {
           for (j = i; j <= n; j++) swap(a[x][j], a[i][j]);
           swap(b[x], b[i]);
       }
       for (j = i + 1; j <= n; j++)
       {
           double y = a[j][i] / a[i][i];
           for (k = i; k <= n; k++) a[j][k] -= a[i][k] * y;
           b[j] -= b[i] * y;
       }
   }
   for (i = n; i >= 1; i--)
   {
       ans[i] = b[i];
       for (j = i + 1; j <= n; j++) ans[i] -= a[i][j] * ans[j];
       ans[i] /= a[i][i];
   }
}

inline int calc1(int i, int l, int r)
{
   int res = s1[i][r];
   res -= (ll)s1[i][l - 1] * p1[r - l + 1] % h1;
   if (res < 0) res += h1;
   return res;
}

inline int calc2(int i, int l, int r)
{
   int res = s2[i][r];
   res -= (ll)s2[i][l - 1] * p2[r - l + 1] % h2;
   if (res < 0) res += h2;
   return res;
}

int main()
{
   scanf("%d %d\n", &n, &m);
   int i, j, k; 
   for (i = 1; i <= n; i++) scanf("%s", s[i] + 1);
   p[0] = p1[0] = p2[0] = 1; n++;
   for (i = 1; i <= m; i++) 
   p[i] = 0.5 * p[i - 1], p1[i] = z1 * p1[i - 1] % h1, p2[i] = z2 * p2[i - 1] % h2;
   for (i = 1; i < n; i++)
   for (j = 1; j <= m; j++)
   s1[i][j] = (z1 * s1[i][j - 1] + s[i][j]) % h1, s2[i][j] = (z2 * s2[i][j - 1] + s[i][j]) % h2;
   for (i = 1; i < n; i++)
   for (j = 1; j < n; j++)
   {
       for (k = 1; k <= m; k++)
       if (calc1(i, 1, k) == calc1(j, m - k + 1, m) && calc2(i, 1, k) == calc2(j, m - k + 1, m))
       {
           if (k == m) a[i][j]++;
           else a[i][j] += p[m - k];
       }
       a[i][n] = -0.125; a[n][i] = 1;
   } 
   b[n] = 1.0;
   gauss();
   for (i = 1; i < n; i++) printf("%.7lf\n", ans[i]);
   return 0;
}