【洛谷】P8775 [蓝桥杯 2022 省 A] 青蛙过河 的题解

【洛谷】P8775 [蓝桥杯 2022 省 A] 青蛙过河 的题解

题目传送门

思路

题意很清晰，需要注意的点是，青蛙要往返 $2x$ 次，石头的下标从 $h_1\sim h_{n-1}$，左岸看作 $h_0$、右岸 $h_n$。

• 首先很显然的一眼二分答案。二分青蛙的跳跃能力，找到最大的能过河的情况。

• 难点在于 check 函数如何处理。需要基于贪心先得出几个结论：

1. 不管跳跃能力为多少，青蛙在 $i$ 位都会尽可能往右跳，如果 $j$ 位承载不下了，就让 $j-1$ 位，$j-2$ 位承载 $\dots$

2. 往返 $2x$ 次，跟从左到右跳 $2x$ 次本质是一样的。

• 我们只需要用一个数组记录每个石头位置能承载的最大跳跃次数（在自身高度限制下），for 循环对于每个 $i$ 往右边传递自身的承载。传递到最后，右岸 $h_n$ 肯定会有一个承载值，判断该值是否为 $2x$，这就是我们的二分判断了。

• 但根据我们之前的跳跃模拟，是一个双重 for 循环的结构，这样显然会超时。可以注意到，对于任意一个位置 $i$ ，一旦它的高度变成 $0$ 了，之后怎么也不会选择它作为落脚点，我们可以用数据结构，这里是并查集，将该点往前路径压缩，这样就避免了 $O(n^2)$ 的查找，每个点最多被遍历一次。

代码

#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;
}