CF573E Bear and Bowling 题解

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

26.CF573E Bear and Bowling 题解2024-08-06

27.CF578F Mirror Box 题解2024-08-2228.[AGC038E] Gachapon 题解02-1929.CF666E Forensic Examination 题解02-2030.CF671E Organizing a Race 题解02-24

收起

Description

• 给定一个长度为 $n$ 的序列 $a_{1\dots n}$。
• 你要求一个 $a$ 的子序列 $b_{1\dots m}$（可以为空），使得 $\sum _{i=1}^{m}i{b}_i$ 的值最大。
• $n\le {10}^5$，$|a_i|\le {10}^7$。

Solution

有一个显然的 dp 是设 $f_{i,j}$ 表示前 $i$ 个数，选 $j$ 个数的最大值，转移即为：$f_{i,j}=max\{f_{i-1,j},f_{i-1,j-1}+j\cdot a_i\}$，由于这题时限很大并且 $n$ 很小，所以这个能过。。。

考虑优化。

有一个结论是对于每个 $i$，都存在一个分界点 $k_i$，使得对于 $\mathrm{\forall }j<k_i$，$f_{i,j}=f_{i-1,j}$，对于 $j\ge k_i$，$f_{i,j}=f_{i-1,j-1}+j\cdot a_i$。

证明就考虑设 $g_{i,j}=f_{i,j}-f_{i,j-1}$，那么 $f_{i,j}=f_{i-1,j}$ 的充分必要条件为 $f_{i-1,j}\ge f_{i-1,j-1}+j\cdot a_i$，即 $\frac{g_{i-1,j}}{j}\ge a_i$。

通过观察可以发现 $\frac{g_{i,j}}{j}$ 对于 $j$ 单调不增，那么不妨假设 $\frac{g_{i-1,j}}{j}$ 单调不增，考虑归纳证明 $g_i$ 也满足条件。

设 $k$ 为满足 $\frac{g_{i-1,j}}{j}<a_i$ 的最小的 $j$，则对于 $j\in [0,k-1]$，$g_{i,j}=g_{i-1,j}$。同时 $g_{i,k}$ 变为 $k\cdot a_i$，所以 $\frac{g_{i,k}}{k}=a_i\le \frac{g_{i,k-1}}{k-1}$。

对于 $j>k$，$g_{i,j}=g_{i-1,j-1}+a_i$，所以对于 $j>k$ 满足 $\frac{g_{i,j}}{j}\ge \frac{g_{i,j+1}}{j+1}$ 的条件为：

$$\begin{array}{rl}\frac{g_{i-1,j-1}+a_i}{j}& \ge \frac{g_{i-1,j}+a_i}{j+1}\\ g_{i-1,j-1}& \ge \frac{j}{j+1}\cdot g_{i-1,j}-\frac{1}{j+1}\cdot a_i\\ g_{i-1,j-1}-(\frac{j}{j+1}\cdot g_{i-1,j}-\frac{1}{j+1}\cdot a_i)& \ge 0\end{array}$$

又因为：

$$\begin{array}{rl}LHS\ge & \frac{j-1}{j}\cdot g_{i-1,j}-\frac{j}{j+1}\cdot g_{i-1,j}+\frac{1}{j+1}\cdot a_i\\ =& \frac{j\cdot a_i-g_{i-1,j}}{j(j+1)}\\ \ge & 0\end{array}$$

所以结论得证。

然后就可以从前往后枚举 $a_i$，维护 $g$ 数组，每次相当于是在某个位置插入和将某个后缀区间加某个数，并且需要快速找到分界点，可以用平衡树维护。

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

Code

#include <bits/stdc++.h>
 
// #define int int64_t
 
const int kMaxN = 1e5 + 5;
 
int n;
int64_t a[kMaxN];
 
struct FHQTreap {
  int tot = 0, rt, ch[kMaxN][2], rd[kMaxN], sz[kMaxN];
  int64_t val[kMaxN], rnk[kMaxN], tag1[kMaxN], tag2[kMaxN];
  std::mt19937 rnd;
 
  int newnode(int64_t x, int64_t y) {
    val[++tot] = x, rnk[tot] = y;
    ch[tot][0] = ch[tot][1] = tag1[tot] = tag2[tot] = 0, rd[tot] = rnd();
    sz[tot] = 1;
    return tot;
  }
 
  FHQTreap() {
    tot = 0, rnd.seed(std::random_device{}());
    rt = newnode(-1e18, 1);
  }
 
  void pushup(int x) {
    sz[x] = sz[ch[x][0]] + sz[ch[x][1]] + 1;
  }
 
  void addtag(int x, int64_t v1, int64_t v2) {
    val[x] += v1, rnk[x] += v2, tag1[x] += v1, tag2[x] += v2;
  }
 
  void pushdown(int x) {
    if (ch[x][0]) addtag(ch[x][0], tag1[x], tag2[x]);
    if (ch[x][1]) addtag(ch[x][1], tag1[x], tag2[x]);
    tag1[x] = tag2[x] = 0;
  }
 
  int merge(int x, int y) {
    if (!x || !y) return x + y;
    pushdown(x), pushdown(y);
    if (rd[x] < rd[y]) {
      ch[x][1] = merge(ch[x][1], y), pushup(x);
      return x;
    } else {
      ch[y][0] = merge(x, ch[y][0]), pushup(y);
      return y;
    }
  }
 
  void split(int x, int v, int &a, int &b) { // a : >= v, b : < v
    if (!x) return void(a = b = 0);
    pushdown(x);
    if (val[x] >= rnk[x] * v) {
      a = x, split(ch[x][1], v, ch[a][1], b);
    } else {
      b = x, split(ch[x][0], v, a, ch[b][0]);
    }
    pushup(x);
  }
 
  void ins(int64_t x) {
    int a, b;
    split(rt, x, a, b);
    if (b) addtag(b, x, 1);
    rt = merge(merge(a, newnode(x * (sz[a] + 1), sz[a] + 1)), b);
  }
 
  int64_t getval(int x) {
    if (!x) return 0;
    pushdown(x);
    int64_t ret = std::max<int64_t>(val[x], 0);
    return ret + getval(ch[x][0]) + getval(ch[x][1]);
  }
} t;
 
void dickdreamer() {
  std::cin >> n;
  for (int i = 1; i <= n; ++i) {
    std::cin >> a[i];
    t.ins(a[i]);
  }
  std::cout << t.getval(t.rt) << '\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;
}