Codeforces-1700 D: River Locks 【动态规划、二分查找】

Codeforces-1700 D: River Locks

题目传送门：Codeforces-1700

题目

题目截图

样例描述

题目大意

  直线上有一个长度为 $n$ 的水坝，水坝上每个单位长度上有一个容器，第 $i$ 个容器可以存储 $v_i$ 的容器，每个容器上面有一个水管，一个水管打开后会向下面的容器注入 $1$ 单位的水，当 $i$ 容器被注满水后，新增的水会瞬间转移到 $i+1$ 容器中。若最后一个容器被注满水，再向该容器注水，多出的水会流入河流。
  现在给 $q$ 组询问，对于每组询问，会得到一个时间 $t$，问最少打开多少个管道，才能在 $t$ 时间内将所有的容器注满。

题目解析

  就不能看着那个图想这道题。。因为我想着想着就变成因为高度，可能存在一些情况，前面的水留不到后面了，但实际上，不管它们什么形状，题目只表明水会从 $i$ 转移到 $i+1$，因此我们只需要顺序的想就可以了。（其实考虑高度也可以做，用单调栈分成几段，再按下面的方法处理每一段就行）
  首先，如果只开一个管子，显然我们要开第一个。如果开第二个，加速让水注满，那么开第二个显然是更优的，因为同样是开两个管子，我们注水影响的容器数量变多了，但相对于总容量，我们的注水速度没变。那么如果要开 $k$ 个管子，我们肯定要开前 $k$ 个。
  如果我们知道前 $k$ 个管子打开，需要多少时间才能注满全部的容器，那么我们就可以用二分处理每一个询问了。那么我们很容易想到，打开管子的容器和没有打开管子的容器需要分开讨论，如果前面的容器注满水，那么后面容器何时被注满水也只是简单计算，因此，前几个容器何时全部被注满水是必须讨论的点。
  设使用前 $i$ 个管子注满前 $i$ 个容器所需时间为 $d{p}_i$，容量的前缀和为 $pr{e}_i$，首先直接计算 $d{p}_i$ 是不行的，因为我们无法保证前面的所有容器均可以被注满。那么我们可以用动态规划去保留能够让 $i-1$ 注满水的性质：
d p i = max ⁡ { d p i − 1 , ⌈ p r e i i ⌉ } dp_i = \max \{dp_{i-1}, \lceil \frac{pre_i }{ i}\rceil \} dpi​=max{dpi−1​,⌈iprei​​⌉}
  能够这样做是因为，若 $\lceil \frac{pr{e}_i}{i}\rceil >d{p}_{i-1}$，说明在 $\lceil \frac{pr{e}_i}{i}\rceil$ 的时间下，首先满足 $i-1$ 个容器被充满，之后我们就可以填第 $i$ 个容器了；反之则说明只开前 $i-1$ 个容器打开，足够填充前 $i$ 个容器了。
  在知道了 $d{p}_i$ 之后，类似地，我们就可以知道打开前 $i$ 个管子，需要多少时间才能注满所有容器了：
$$\max (d{p}_i,\lceil \frac{pr{e}_n}{i}\rceil )$$
  之后我们直接用二分来找到满足条件的最小管子数量就可以了。
  这种取最大值保留性质的问题非常经典，在这个问题中，只有时间一个维度，因此在保留前面容量充满和前面容器加当前容器充满中选最大的一个就可以同时满足它们的要求。这种思路常见有趣，是一种经典问题。

Code

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn = 2e5 + 7;
LL v[maxn], p[maxn];

int main() {
   int n, q, t;
   cin >> n;
   for(int i=1; i<=n; ++i)
       cin >> v[i], v[i] += v[i-1];
   for(int i=1; i<=n; ++i)
       p[i] = max((v[i] + i - 1) / i, p[i-1]);
   for(int i=1; i<=n; ++i)
       p[i] = max(p[i], (v[n] + i - 1) / i);
   reverse(p+1, p+n+1);

   cin >> q;
   while(q--) {
       cin >> t;
       int idx = upper_bound(p+1, p+n+1, t) - p - 1;
       idx = n - idx + 1;
       if (idx > n) cout << -1 << endl; else cout << idx << endl;
   }
   return 0;
}