[AGC038E] Gachapon 题解

合集 - IOI2020 国家集训队作业 题解(30)

1.CF505E Mr. Kitayuta vs. Bamboos 题解2024-07-042.CF506E Mr. Kitayuta's Gift 题解2024-07-223.CF512D Fox And Travelling 题解2024-07-224.CF516D Drazil and Morning Exercise 题解2024-02-085.CF516E Drazil and His Happy Friends 题解2024-07-236.CF521D Shop 题解2024-07-237.CF521E Cycling City 题解2024-07-238.CF527E Data Center Drama 题解2024-07-249.CF547D Mike and Fish 题解2024-07-2410.CF559E Gerald and Path 题解2024-07-2511.CF568C New Language 题解2024-07-2512.CF576D Flights for Regular Customers 题解2024-07-2613.CF578E Walking! 题解2024-07-2614.CF585F Digits of Number Pi 题解2024-07-2615.CF605E Intergalaxy Trips 题解2024-07-2716.CF613E Puzzle Lover 题解2024-07-2717.CF626G Raffles 题解2024-07-2818.CF526G Spiders Evil Plan 题解2024-07-2819.CF538G Berserk Robot 题解2024-07-2920.CF538H Summer Dichotomy 题解2024-07-3021.CF553E Kyoya and Train 题解2024-08-0122.CF566C Logistical Questions 题解2024-08-0223.CF566E Restoring Map 题解2024-08-0424.CF568E Longest Increasing Subsequence 题解2024-08-0625.CF571E Geometric Progressions 题解2024-08-0626.CF573E Bear and Bowling 题解2024-08-0627.CF578F Mirror Box 题解2024-08-22

28.[AGC038E] Gachapon 题解02-19

29.CF666E Forensic Examination 题解02-2030.CF671E Organizing a Race 题解02-24

收起

Description

有一个随机数生成器，生成 $[0,n-1]$ 之间的整数，其中生成 $i$ 的概率为 $\frac{A_i}{S}$，其中，$S=\sum A_i$。

这个随机数生成器不断生成随机数，当 $\mathrm{\forall }i\in [0,n-1]$，$i$ 至少出现了 $B_i$ 次时，停止生成，否则继续生成。

求期望生成随机数的次数，输出答案对 $998244353$ 取模的结果。

$A_i,B_i\ge 1$，$\sum A_i,\sum B_i,n\le 400$。

Solution

显然先要 min-max 容斥。

对于集合 $T$，有 $P=\frac{\sum _{i\in T}{A}_i}{S}$ 的概率选到 $T$ 内的点，所以期望 $\frac{1}{P}$ 次选到 $T$ 内，$T$ 对答案的贡献即为让 $T$ 中存在一个 $i$ 至少出现 $B_i$ 次的期望操作数乘 $\frac{1}{P}$。

考虑枚举还没满足条件的每个时刻每个位置被选的次数，设分别选了 $C_1,C_2,\dots ,C_m$ 次。

首先需要满足 $C_i<B_i$，然后用总方案数除以得到这种状态的方案数得到获得此状态的概率，即 $\frac{(\sum C_i)!}{\prod C_i!}\cdot \prod {\left(\frac{A_i}{\sum _{i\in T}{A}_i}\right)}^{C_i}$。

所以最终答案为：

$$\begin{array}{rl}ans& =\sum _{T\subseteq \{1,2,\dots ,n\}}(-1{)}^{|T|-1}\cdot \frac{S}{\sum _{i\in T}{A}_i}\cdot \sum _{\mathrm{\forall }i\in T,C_i<B_i}\frac{(\sum C_i)!}{\prod C_i!}\cdot \prod {\left(\frac{A_i}{\sum _{i\in T}{A}_i}\right)}^{C_i}\end{array}$$

dp 时记录选的 $\sum A_i$ 和 $\sum C_i$ 即可。

时间复杂度：$O(\sum A_i\cdot {(\sum B_i)}^2)$。

Code

#include <bits/stdc++.h>
 
// #define int int64_t
 
const int kMaxN = 405, kMod = 998244353;
 
int n, sa, sb;
int a[kMaxN], b[kMaxN], fac[kMaxN], ifac[kMaxN], f[kMaxN][kMaxN];
 
constexpr int qpow(int bs, int64_t idx = kMod - 2) {
  int ret = 1;
  for (; idx; idx >>= 1, bs = (int64_t)bs * bs % kMod)
    if (idx & 1)
      ret = (int64_t)ret * bs % kMod;
  return ret;
}
 
inline int add(int x, int y) { return (x + y >= kMod ? x + y - kMod : x + y); }
inline int sub(int x, int y) { return (x >= y ? x - y : x - y + kMod); }
inline void inc(int &x, int y) { (x += y) >= kMod ? x -= kMod : x; }
inline void dec(int &x, int y) { (x -= y) < 0 ? x += kMod : x; }
 
void prework(int n = 400) {
  fac[0] = ifac[0] = 1;
  for (int i = 1; i <= n; ++i) {
    fac[i] = 1ll * i * fac[i - 1] % kMod;
    ifac[i] = qpow(fac[i]);
  }
}
 
void dickdreamer() {
  std::cin >> n;
  for (int i = 1; i <= n; ++i)
    std::cin >> a[i] >> b[i];
  prework();
  f[0][0] = kMod - 1;
  for (int i = 1; i <= n; ++i) {
    for (int j = sa; ~j; --j) {
      for (int k = sb; ~k; --k) {
        if (!f[j][k]) continue;
        int pw = 1;
        for (int c = 0; c <= b[i] - 1; ++c) {
          dec(f[j + a[i]][k + c], 1ll * f[j][k] * ifac[c] % kMod * pw % kMod);
          pw = 1ll * pw * a[i] % kMod;
        }
      }
    }
    sa += a[i], sb += b[i];
  }
  int ans = 0;
  for (int i = 1; i <= sa; ++i) {
    for (int j = 0; j <= sb; ++j)
      inc(ans, 1ll * sa * qpow(i) % kMod * fac[j] % kMod * qpow(qpow(i), j) % kMod * f[i][j] % kMod);
  }
  std::cout << ans << '\n';
}
 
int32_t main() {
#ifdef ORZXKR
  freopen("in.txt", "r", stdin);
  freopen("out.txt", "w", stdout);
#endif
  std::ios::sync_with_stdio(0), std::cin.tie(0), std::cout.tie(0);
  int T = 1;
  // std::cin >> T;
  while (T--) dickdreamer();
  // std::cerr << 1.0 * clock() / CLOCKS_PER_SEC << "s\n";
  return 0;
}