给你一个浮点数 hour
,表示你到达办公室可用的总通勤时间。要到达办公室,你必须按给定次序乘坐 n
趟列车。另给你一个长度为 n
的整数数组 dist
,其中 dist[i]
表示第 i
趟列车的行驶距离(单位是千米)。
每趟列车均只能在整点发车,所以你可能需要在两趟列车之间等待一段时间。
- 例如,第
1
趟列车需要1.5
小时,那你必须再等待0.5
小时,搭乘在第 2 小时发车的第2
趟列车。
返回能满足你准时到达办公室所要求全部列车的 最小正整数 时速(单位:千米每小时),如果无法准时到达,则返回 -1
。
生成的测试用例保证答案不超过 107
,且 hour
的 小数点后最多存在两位数字 。
示例 1:
输入:dist = [1,3,2], hour = 6 输出:1 解释:速度为 1 时: - 第 1 趟列车运行需要 1/1 = 1 小时。 - 由于是在整数时间到达,可以立即换乘在第 1 小时发车的列车。第 2 趟列车运行需要 3/1 = 3 小时。 - 由于是在整数时间到达,可以立即换乘在第 4 小时发车的列车。第 3 趟列车运行需要 2/1 = 2 小时。 - 你将会恰好在第 6 小时到达。
示例 2:
输入:dist = [1,3,2], hour = 2.7 输出:3 解释:速度为 3 时: - 第 1 趟列车运行需要 1/3 = 0.33333 小时。 - 由于不是在整数时间到达,故需要等待至第 1 小时才能搭乘列车。第 2 趟列车运行需要 3/3 = 1 小时。 - 由于是在整数时间到达,可以立即换乘在第 2 小时发车的列车。第 3 趟列车运行需要 2/3 = 0.66667 小时。 - 你将会在第 2.66667 小时到达。
示例 3:
输入:dist = [1,3,2], hour = 1.9 输出:-1 解释:不可能准时到达,因为第 3 趟列车最早是在第 2 小时发车。
解法:二分答案
由于除了最后一趟列车,前面的每趟列车至少花费 1 小时,且最后一趟列车花费的时间大于 0,因此 hour 必须严格大于 n−1。
class Solution {
public int minSpeedOnTime(int[] dist, double hour) {
int n = dist.length;
if (hour <= n - 1) {
return -1;
}
// 取dist[0,...,n - 2]的最大值
int max = 0;
for (int i = 0; i <= n - 2; i++) {
max = Math.max(max, dist[i]);
}
// 闭区间写法
int left = 1;
int right = Math.max(max, (int) Math.ceil( dist[n - 1] / (hour - n + 1) ) );
while (left <= right) {
// 循环不变量
// check(left - 1), false
// check(right + 1), true
int mid = left + (right - left) / 2;
if (check(dist, mid, hour)) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return left;
// 左闭右开区间写法
// int left = 1;
// int right = Math.max(max, (int) Math.ceil( dist[n - 1] / (hour - n + 1) ) ) + 1;
// while (left < right) {
// // 循环不变量
// // check(left - 1), false
// // check(right), true
// int mid = left + (right - left) / 2;
// if (check(dist, mid, hour)) {
// right = mid;
// } else {
// left = mid + 1;
// }
// }
// return left;
// 左开右闭区间写法
// int left = 0;
// int right = Math.max(max, (int) Math.ceil( dist[n - 1] / (hour - n + 1) ) );
// while (left < right) {
// // 循环不变量
// // check(left), false
// // check(right + 1), true
// int mid = left + (right - left + 1) / 2;
// if (check(dist, mid, hour)) {
// right = mid - 1;
// } else {
// left = mid;
// }
// }
// return right + 1;
// 开区间写法
// int left = 0;
// int right = Math.max(max, (int) Math.ceil( dist[n - 1] / (hour - n + 1) ) ) + 1;
// while (left + 1 < right) {
// // 循环不变量
// // check(left), false
// // check(right), true
// int mid = left + (right - left) / 2;
// if (check(dist, mid, hour)) {
// right = mid;
// } else {
// left = mid;
// }
// }
// return right;
}
private boolean check(int[] dist, int x, double hour) {
double sum = 0;
for (int i = 0; i < dist.length; i++) {
if (i == dist.length - 1) {
sum += (double) dist[i] / x;
break;
}
// dist[i] / x 上取整
sum += (dist[i] + x - 1) / x;
}
if (sum <= hour) {
return true;
}
return false;
}
}
复杂度分析
时间复杂度:O(n*log U),n为 dist 的长度,U = (Math.max(max, (int) Math.ceil( dist[n - 1] / (hour - n + 1) ) ); 所以总的时间复杂度为O(n - 1 + n*log U ) = O(n*log U)。
空间复杂度:O(1)。