题目描述
一年一度的“跳石头”比赛又要开始了!这项比赛将在一条笔直的河道中进行,河道中分布着一些巨大岩石。组委会已经选择好了两块岩石作为比赛起点和终点。在起点和终点之间,有 N N N 块岩石(不含起点和终点的岩石)。
在比赛过程中,选手们将从起点出发,每一步跳向相邻的岩石,直至到达终点。
为了提高比赛难度,组委会计划移走一些岩石,使得选手们在比赛过程中的最短跳跃距离尽可能长。
由于预算限制,组委会至多从起点和终点之间移走 M M M 块岩石(不能移走起点和终点的岩石)。
输入格式
输入文件第一行包含三个整数
L
,
N
,
M
L,N,M
L,N,M,分别表示起点到终点的距离,起点和终点之间的岩石数,以及组委会至多移走的岩石数。
接下来 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块岩石与起点的距离。
这些岩石按与起点距离从小到大的顺序给出,且不会有两个岩石出现在同一个位置。
输出格式
输出文件只包含一个整数,即最短跳跃距离的最大值。
数据范围
0
≤
M
≤
N
≤
50000
0≤M≤N≤50000
0≤M≤N≤50000,
1
≤
L
≤
1
0
9
1≤L≤10^9
1≤L≤109
输入样例
25 5 2
2
11
14
17
21
输出样例
4
算法思想(二分查找)
根据题目描述,要在 n n n块石头中去掉 m m m块,使得在比赛过程中的最短跳跃距离尽可能长。
一种朴素的思想是进行 m m m次循环,每次去掉一块距离最短的石头,最后再求最短的距离。实现比较复杂,而且时间复杂度为 O ( m × n ) O(m\times n) O(m×n),结果一定超时。
由于最短跳跃距离在范围 [ 0 , L ] [0, L] [0,L]之间,这里可以考虑使用二分查找来确定最短跳跃距离 x x x。
- 在查找过程中,需要遍历每块石头,如果第
i
块石头距离上块石头的距离小于 x x x,那么需要将第i
块石头移走。这样可以统计跳出最短距离 x x x时,需要移走多少块石头 s u m sum sum - 如果 s u m sum sum不超过 m m m时,则 x x x为符合条件的值。
- 继续查找有无更长的 x x x可以满足要求。
时间复杂度
二分查找的时间复杂度为 O ( l o g L ) O(logL) O(logL),检查 x x x的时间复杂度为 O ( n ) O(n) O(n),总的时间复杂度为 O ( n × l o g L ) ≈ O ( 30 × 50000 ) O(n\times logL)≈O(30\times50000) O(n×logL)≈O(30×50000)。
代码实现
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int a[N];
int l, n, m;
int check(int x)
{
int last = 0, sum = 0;
//注意,需要枚举到终点
for(int i = 0; i <= n; i ++)
{
if(a[i] - last < x) sum ++;
else last = a[i];
}
return sum <= m;
}
int main()
{
scanf("%d%d%d", &l, &n, &m);
for(int i = 0; i < n; i ++) scanf("%d", a + i);
a[n] = l; //注意需要判断终点
int L = 0, R = l;
while(L < R)
{
//查找符合条件的最大值,适用二分查找模板二
int mid = L + R + 1 >> 1;
if(check(mid)) L = mid;
else R = mid - 1;
}
printf("%d\n", L);
return 0;
}