【ZJOI2019】麻将

· · 题解

考虑给你一副牌，如何判断这副牌是否是胡的。

容易发现相同的顺子不会选三个以上，于是考虑花色从小到大进行dp。令dp_{0/1, i, j, k}表示是否有对子，考虑了前i种花色的牌，以花色i - 1的牌为开头的顺子准备选j个，以花色i的牌为开头的顺子准备选k个，此时能选的最大的面子数。转移只要枚举以花色i + 1的牌开头的顺子准备选几个，剩下的牌组成刻子就行了(具体细节可以看代码中结构体State中的Trans函数)。

判定胡牌的条件有两种：\exists (j,k), dp_{1, n, j, k} \ge 4；记cnt为牌数\ge 2的花色数，cnt \ge 7。

可以发现这个dp和状态i这一维关系不是很大，dp_{0}和dp_{1}的转移方式也都是一样的。只考虑有dp_{*,j,k}这个3 \times 3的矩阵，其转移本质上是接收一个数字（表示下一种花色的牌的数量），然后变成了另一个3 \times 3的矩阵，我们可以把它看成一个自动机的模型。事实上这样的dp状态（3 \times 3的矩阵）并不很多。我们用结构体State来形容这个矩阵。为了表示整个当前胡牌状态，我们需要两个State分别表示有没有选对子（即dp_0和dp_1），以及cnt。我们把这三个东西放到结构体Mahjong中，我们很容易可以得到Mahjong的Trans关系。经搜索Mahjong的状态数总共有3956个。

回过来看我们要算的答案：ans = \sum\limits_{a = 13}^{4n} p(a)，其中p(a)表示你总共摸了a张牌之后仍然不能胡的概率。于是我们只要算摸了a张牌仍然不能胡的排列数即可。令f_{i, j, k}表示我们考虑了前i种花色的牌，当前胡牌状态为j（j是一个Mahjong类），已经摸了k张牌的排列数有多少。转移枚举摸了花色i + 1的牌数z，则由f_{i, j, k}转移到f_{i + 1, trans(j,z), k + z}，乘上的系数就是(4 - org_{i + 1})^{\underline{z - org{i + 1}}} \binom{k + z - sum_{i + 1}}{z - org_{i + 1}}。org_i表示原有的13张牌中花色为i的有几张，sum则是org的前缀和，这个式子的意思就是我们需要在没被选过的4 - org_{i + 1}张牌中选z - org_{i + 1}张的排列，并且插入到之前的排列中，但前13张牌的顺序是固定的。

最后p(a)是很好算的，p(a) = \frac{\sum\limits_{j \; can \; not \; win } f_{n, j, a}}{ (4n - 13)^{\underline{a - 13}} }。


#include <bits/stdc++.h>

using namespace std;

const int N = 1e2 + 5;
const int M = 4e3 + 5;
const int MOD = 998244353;

namespace {
  int ch[M][5], fac[5];
  int Add(int a, int b) { return (a += b) >= MOD? a - MOD : a; }
  void Upd(int &a, int b) { a = Add(a, b); }
  int Mul(int a, int b) { return (long long)a * b % MOD; }
  int Inv(int x) { return (x == 1)? 1 : Mul(MOD - MOD / x, Inv(MOD % x)); }
  void Math_init() {
    for (int i = 0; i < M; ++i) {
      ch[i][0] = 1;
      for (int j = 1; j <= min(i, 4); ++j) {
        ch[i][j] = Add(ch[i - 1][j - 1], ch[i - 1][j]);
      }
    }
    fac[0] = 1;
    for (int i = 1; i < 5; ++i) {
      fac[i] = Mul(fac[i - 1], i);
    }
  }
}

bool Chkmax(int &a, int b) {
  return (a < b)? (a = b, 1) : (0);
}

struct State {
  int dp[3][3]; // (last last bar, last bar)
  State() {
    memset(dp, -1, sizeof dp);
  }

  friend bool operator < (State a, State b) {
    for (int i = 0; i < 3; ++i)
      for (int j = 0; j < 3; ++j)
        if (a.dp[i][j] != b.dp[i][j])
          return a.dp[i][j] < b.dp[i][j];
    return 0;
  }

  friend State Max(State a, State b) {
    for (int i = 0; i < 3; ++i) 
      for (int j = 0; j < 3; ++j)
        a.dp[i][j] = max(a.dp[i][j], b.dp[i][j]);
    return a;
  }

  friend State Trans(State a, int b) {
    State c;
    for (int i = 0; i < 3; ++i)
      for (int j = 0; j < 3; ++j)
        if (~a.dp[i][j])
          for (int k = 0; k < 3 && i + j + k <= b; ++k)
            Chkmax(c.dp[j][k], min(i + a.dp[i][j] + (b - i - j - k) / 3, 4));
    return c;
  }

};

struct Mahjong {
  pair<State, State> god;
  int cnt;
  Mahjong() {
    memset(god.first.dp, -1, sizeof god.first.dp);
    memset(god.second.dp, -1, sizeof god.second.dp);
    god.first.dp[0][0] = cnt = 0;
  }

  friend bool operator < (Mahjong a, Mahjong b) {
    return a.cnt != b.cnt? a.cnt < b.cnt : a.god < b.god;
  }

  friend Mahjong Trans(Mahjong a, int b) {
    a.cnt = min(a.cnt + (b >= 2), 7);
    a.god.second = Trans(a.god.second, b);
    if (b >= 2) {
      a.god.second = Max(a.god.second, Trans(a.god.first, b - 2));
    }
    a.god.first = Trans(a.god.first, b);
    return a;
  }

  bool right() {
    if (cnt == 7) return 1;
    for (int i = 0; i < 3; ++i)
      for (int j = 0; j < 3; ++j)
        if (god.second.dp[i][j] == 4) return 1;
    return 0;
  }

} mahjong[M];

int n, tot;
map<Mahjong, int> idx;
bool win[M];
int org[N], f[N][M][4 * N], trans[M][5];

void Dfs_mahjong(Mahjong now) {
  if (idx.find(now) != idx.end()) return;
  mahjong[++tot] = now;
  win[tot] = now.right();
  idx[now] = tot;
  for (int i = 0; i <= 4; ++i) {
    Dfs_mahjong(Trans(now, i));
  }
}

int main() {
  Math_init();
  Dfs_mahjong(Mahjong());

  for (int i = 1; i <= tot; ++i) {
    for (int j = 0; j <= 4; ++j) {
      trans[i][j] = idx[Trans(mahjong[i], j)];
    }
  }

  scanf("%d", &n);
  for (int i = 0, x; i < 13; ++i) {
    scanf("%d%*d", &x);
    ++org[x];
  }

  f[0][1][0] = 1;
  for (int i = 0, cp = 0; i < n; ++i) { // consider 1 ... i
    cp += org[i + 1];
    for (int j = 1; j <= tot; ++j) { // mahjong j
      for (int l = org[i + 1]; l <= 4; ++l) { // trans
        int *nf = f[i + 1][trans[j][l]], *ff = f[i][j];
        int tmp = Mul(ch[4 - org[i + 1]][l - org[i + 1]], fac[l - org[i + 1]]);
        for (int k = 0; k + l <= 4 * n; ++k) { // have chosen k cards
          if (!ff[k]) continue;
          Upd(nf[k + l], Mul(ff[k], Mul(ch[k + l - cp][l - org[i + 1]], tmp)));
        }
      }
    }
  }

  int ans = 0, dw = 1;
  for (int i = 13; i <= 4 * n; ++i) {
    int up = 0;
    for (int j = 1; j <= tot; ++j) {
      if (!win[j]) Upd(up, f[n][j][i]);
    }
    Upd(ans, Mul(up, Inv(dw)));
    dw = Mul(dw, 4 * n - i);
  }
  printf("%d\n", ans);

  return 0;
}