题目背景
一年一度的“跳石头”比赛又要开始了!
题目描述
这项比赛将在一条笔直的河道中进行,河道中分布着一些巨大岩石。组委会已经选择好了两块岩石作为比赛起点和终点。在起点和终点之间,有 N N N 块岩石(不含起点和终点的岩石)。在比赛过程中,选手们将从起点出发,每一步跳向相邻的岩石,直至到达终点。
为了提高比赛难度,组委会计划移走一些岩石,使得选手们在比赛过程中的最短跳跃距离尽可能长。由于预算限制,组委会至多从起点和终点之间移走 M M M 块岩石(不能移走起点和终点的岩石)。
输入格式
第一行包含三个整数 L , N , M L,N,M L,N,M,分别表示起点到终点的距离,起点和终点之间的岩石数,以及组委会至多移走的岩石数。保证 L ≥ 1 L \geq 1 L≥1 且 N ≥ M ≥ 0 N \geq M \geq 0 N≥M≥0。
接下来 N N N 行,每行一个整数,第 i i i 行的整数 D i ( 0 < D i < L ) D_i( 0 < D_i < L) Di(0<Di<L), 表示第 i i i 块岩石与起点的距离。这些岩石按与起点距离从小到大的顺序给出,且不会有两个岩石出现在同一个位置。
输出格式
一个整数,即最短跳跃距离的最大值。
样例 #1
样例输入 #1
25 5 2
2
11
14
17
21
样例输出 #1
4
提示
输入输出样例 1 说明
将与起点距离为 2 2 2和 14 14 14 的两个岩石移走后,最短的跳跃距离为 4 4 4(从与起点距离 17 17 17 的岩石跳到距离 21 21 21 的岩石,或者从距离 21 21 21 的岩石跳到终点)。
数据规模与约定
对于
20
%
20\%
20%的数据,
0
≤
M
≤
N
≤
10
0 \le M \le N \le 10
0≤M≤N≤10。
对于
50
%
50\%
50% 的数据,
0
≤
M
≤
N
≤
100
0 \le M \le N \le 100
0≤M≤N≤100。
对于
100
%
100\%
100%的数据,
0
≤
M
≤
N
≤
50000
,
1
≤
L
≤
1
0
9
0 \le M \le N \le 50000,1 \le L \le 10^9
0≤M≤N≤50000,1≤L≤109。
题目分析
假设L=20,M=3,N=6,且各石头间距离如上图。根据题意即求去掉最合适的3个石头后使得剩下三个石头中相邻石头间距离中最小的距离为最大(和去掉其它不合适的3个石头相比)。根据数据规模为50000,计算机每秒执行大约1亿条指令,估计该题时间复杂度为O(NLogL)或O(N).根据找最合适的石头,可使用贪心法。根据最小距离的最大,使用二分法找距离的边界来满足题意。
首先求解在某个距离下可去掉的石头数
// @Param arr: 记录与起点的距离
// @Param len: 记录arr的长度
// @Param gap: 最小距离
// @Return: 在给定的最小距离下,可以去掉的石头个数
int removeAndCount(int arr[], int len, int gap) {
int prev = arr[0];
int count = 0;
for (int i = 1; i < len; i++) {
if ((arr[i] - prev) >= gap) {
prev = arr[i];
}
else {
count++;
}
}
return count;
}
使用二分法找距离边界,使去掉石头数量等于M时,距离为最大。
int left = 1, right = L, mid;
int c;
while (left < right) {
mid = left + (right - left) / 2;
c = removeAndCount(arr, N+2, mid);
if (c <= M) {
left = mid + 1;
}
else {
right = mid;
}
}
再判断一次,若最后一次是对距离mid做加一操作且仍满足要求,应打印left,否则打印mid。
c = removeAndCount(arr, N + 2, left);
if (c > M) {
left--;
}
最后给出完整代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int removeAndCount(int arr[], int len, int gap) {
int prev = arr[0];
int count = 0;
for (int i = 1; i < len; i++) {
if ((arr[i] - prev) >= gap) {
prev = arr[i];
}
else {
count++;
}
}
return count;
}
int main() {
int L, M, N;
cin >> L >> N >> M;
int* arr = new int[N+2];
for (int i = 1; i <= N; i++) {
cin >> arr[i];
}
arr[0] = 0;
arr[N + 1] = L;
int left = 1, right = L, mid;
int c;
while (left < right) {
mid = left + (right - left) / 2;
c = removeAndCount(arr, N+2, mid);
if (c <= M) {
left = mid + 1;
}
else {
right = mid;
}
}
c = removeAndCount(arr, N + 2, left);
if (c > M) {
left--;
}
cout << left << endl;
return 0;
}