CF626G Raffles 题解

合集 - 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-27

17.CF626G Raffles 题解2024-07-28

18.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-2228.[AGC038E] Gachapon 题解02-1929.CF666E Forensic Examination 题解02-2030.CF671E Organizing a Race 题解02-24

收起

Description

• 有 $n$ 个奖池，第 $i$ 个奖池的奖金是 $p_i$，已经有 $l_i$ 张彩票押在上面。
• 现在你有 $t$ 张彩票，你需要将你的彩票分配到这些奖池中，并且保证你在每个奖池中押的彩票数不能超过该奖池原有的彩票数。
• 若你在第 $i$ 个奖池中押了 $t_i$ 张彩票，则你中奖的概率为 $\frac{t_i}{t_i+l_i}$，若你中奖，你可以获得这个奖池的全部奖金 $p_i$。
• 一共有 $q$ 次事件，每次事件会使某个 $l_i$ 加 $1$ 或减 $1$。
• 你需要在每个事件后求出在最佳方案下你获得的奖金总数的最大期望值。
• $n,t,q\le 2\times {10}^5$，$p_i,l_i\le {10}^3$，答案精度误差 $\le {10}^{-6}$。

Solution

首先单次 $O(t\log n)$ 是很容易的，就先假设所有 $t_i=0$，每次取出让 $t_i\to t_i+1$ 的增量的最大值，由于对于每个 $i$，$t_i$ 越大增量越小，所以每次贪心取最大值是对的。

考虑怎么支持修改。

不妨设修改了 $x$，让 $l_x\leftarrow l_x+1$。

有一个结论是每次选出当前已经选择的 $\mathrm{\Delta }$ 里的最小值和没选的 $\mathrm{\Delta }$ 的最大值，如果这个最小值小于最大值就贪心地替换，替换的次数不超过 $1$ 次。

证明就考虑设 $\mathrm{\Delta }E(k)={\displaystyle \frac{p_x{l}_x}{(l_x+k)(l_x+k+1)}},\mathrm{\Delta }{E}^{\prime }(k)={\displaystyle \frac{p_x(l_x+1)}{(l_x+k+1)(l_x+k+2)}}$。

注意到 $\mathrm{\Delta }{E}^{\prime }(k)={\displaystyle \frac{p_x(l_x+1)}{(l_x+k+1)(l_x+k+2)}}>\mathrm{\Delta }E(k+1)={\displaystyle \frac{p_x{l}_x}{(l_x+k+1)(l_x+k+2)}}$，所以 $x$ 只有可能是选了的最小的增量被替换，所以最多只有一次。

对于 $l_x\leftarrow l_x-1$ 的情况同理可证结论仍成立。

时间复杂度：$O((n+q+t)\log n)$。

Code

#include <bits/stdc++.h>
 
// #define int int64_t
 
using f64 = long double;
 
const int kMaxN = 2e5 + 5;
 
int n, m, q;
f64 ans;
int l[kMaxN], p[kMaxN], t[kMaxN];
std::multiset<std::pair<f64, int>> st[2];
 
f64 getdet(int x, int v) {
  return (f64)p[x] * l[x] / (l[x] + t[x]) / (l[x] + t[x] + v);
}
 
void add(int x) {
  --m;
  st[1].erase({getdet(x, 1), x});
  if (t[x]) st[0].erase({getdet(x, -1), x});
  ans += getdet(x, 1), ++t[x];
  st[0].emplace(getdet(x, -1), x);
  if (t[x] < l[x]) st[1].emplace(getdet(x, 1), x);
}
 
void del(int x) {
  ++m;
  if (t[x] < l[x]) st[1].erase({getdet(x, 1), x});
  st[0].erase({getdet(x, -1), x});
  ans -= getdet(x, -1), --t[x];
  if (t[x]) st[0].emplace(getdet(x, -1), x);
  st[1].emplace(getdet(x, 1), x);
}
 
void dickdreamer() {
  std::cin >> n >> m >> q;
  for (int i = 1; i <= n; ++i) std::cin >> p[i];
  for (int i = 1; i <= n; ++i) std::cin >> l[i];
  for (int i = 1; i <= n; ++i) st[1].emplace(getdet(i, 1), i);
  for (; m && !st[1].empty();) add(st[1].rbegin()->second);
  for (int i = 1; i <= q; ++i) {
    int op, x;
    std::cin >> op >> x;
    if (op == 1) {
      if (t[x]) st[0].erase({getdet(x, -1), x});
      if (t[x] < l[x]) st[1].erase({getdet(x, 1), x});
      ans -= (f64)p[x] * t[x] / (t[x] + l[x]);
      ++l[x];
    } else {
      if (t[x] == l[x]) del(x);
      if (t[x]) st[0].erase({getdet(x, -1), x});
      if (t[x] < l[x]) st[1].erase({getdet(x, 1), x});
      ans -= (f64)p[x] * t[x] / (t[x] + l[x]);
      --l[x];
    }
    ans += (f64)p[x] * t[x] / (t[x] + l[x]);
    if (t[x]) st[0].emplace(getdet(x, -1), x);
    if (t[x] < l[x]) st[1].emplace(getdet(x, 1), x);
    for (; m && !st[1].empty();) add(st[1].rbegin()->second);
    if (!st[0].empty() && !st[1].empty()) {
      auto [det1, j1] = *st[0].begin();
      auto [det2, j2] = *st[1].rbegin();
      if (det1 < det2) del(j1), add(j2);
    }
    std::cout << std::fixed << std::setprecision(10) << 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;
}