【洛谷】P8775 [蓝桥杯 2022 省 A] 青蛙过河 的题解
题目传送门
思路
题意很清晰,需要注意的点是,青蛙要往返 2 x 2x 2x 次,石头的下标从 h 1 ∼ h n − 1 h_1 \thicksim h_{n-1} h1∼hn−1,左岸看作 h 0 h_0 h0、右岸 h n h_n hn。
-
首先很显然的一眼二分答案。二分青蛙的跳跃能力,找到最大的能过河的情况。
-
难点在于 check 函数如何处理。需要基于贪心先得出几个结论:
-
不管跳跃能力为多少,青蛙在 i i i 位都会尽可能往右跳,如果 j j j 位承载不下了,就让 j − 1 j - 1 j−1 位, j − 2 j - 2 j−2 位承载 … \dots …
-
往返 2 x 2x 2x 次,跟从左到右跳 2 x 2x 2x 次本质是一样的。
-
我们只需要用一个数组记录每个石头位置能承载的最大跳跃次数(在自身高度限制下),for 循环对于每个 i i i 往右边传递自身的承载。传递到最后,右岸 h n h_n hn 肯定会有一个承载值,判断该值是否为 2 x 2x 2x,这就是我们的二分判断了。
-
但根据我们之前的跳跃模拟,是一个双重 for 循环的结构,这样显然会超时。可以注意到,对于任意一个位置 i i i ,一旦它的高度变成 0 0 0 了,之后怎么也不会选择它作为落脚点,我们可以用数据结构,这里是并查集,将该点往前路径压缩,这样就避免了 O ( n 2 ) O(n^2) O(n2) 的查找,每个点最多被遍历一次。
代码
#include <bits/stdc++.h>
#define endl "\n"
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
namespace fastIO {
inline int read() {
register int x = 0, f = 1;
register char c = getchar();
while (c < '0' || c > '9') {
if(c == '-') f = -1;
c = getchar();
}
while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
inline void write(int x) {
if(x < 0) putchar('-'),x = -x;
if(x > 9) write(x / 10);
putchar(x % 10 + '0');
return;
}
}
using namespace fastIO;
int n, m, k, x;
ll s[100005], a[100005], h[100005];
int p[100005];
int find(int x) {
return p[x] == x ? p[x] : p[x] = find(p[x]);
}
bool check(int y) {
memset(s, 0, sizeof(s));
s[0] = 2 * x;
for(int i = 1; i < n; i ++) a[i] = h[i];
for(int i = 0; i <= n; i ++) p[i] = i;
for(int i = 0; i < n; i ++) {
int to = i + y;
if(to >= n) {
s[n] += s[i];
}
else {
while(1) {
int mi = min(a[to], s[i]);
s[to] += mi;
s[i] -= mi;
a[to] -= mi;
if(a[to] == 0) {
p[to] = to - 1;
}
if(s[i] == 0) break;
to = find(to - 1);
if(to <= i) break;
}
}
}
return s[n] == 2 * x;
}
int main() {
//freopen(".in","r",stdin);
//freopen(".out","w",stdout);
n = read(), x = read();
for (int i = 1; i < n; i ++) h[i] = read();
int l = 1, r = n;
while (l < r) {
int mid = l + r >> 1;
if(!check(mid)) l = mid + 1;
else r = mid;
}
write(l);
return 0;
}