给你一个数组 time
,其中 time[i]
表示第 i
辆公交车完成 一趟旅途 所需要花费的时间。
每辆公交车可以 连续 完成多趟旅途,也就是说,一辆公交车当前旅途完成后,可以 立马开始 下一趟旅途。每辆公交车 独立 运行,也就是说可以同时有多辆公交车在运行且互不影响。
给你一个整数 totalTrips
,表示所有公交车 总共 需要完成的旅途数目。请你返回完成 至少 totalTrips
趟旅途需要花费的 最少 时间。
示例 1:
输入:time = [1,2,3], totalTrips = 5 输出:3 解释: - 时刻 t = 1 ,每辆公交车完成的旅途数分别为 [1,0,0] 。 已完成的总旅途数为 1 + 0 + 0 = 1 。 - 时刻 t = 2 ,每辆公交车完成的旅途数分别为 [2,1,0] 。 已完成的总旅途数为 2 + 1 + 0 = 3 。 - 时刻 t = 3 ,每辆公交车完成的旅途数分别为 [3,1,1] 。 已完成的总旅途数为 3 + 1 + 1 = 5 。 所以总共完成至少 5 趟旅途的最少时间为 3 。
示例 2:
输入:time = [2], totalTrips = 1 输出:2 解释: 只有一辆公交车,它将在时刻 t = 2 完成第一趟旅途。 所以完成 1 趟旅途的最少时间为 2 。
解法:二分答案
时间越多,可以完成的旅途也就越多,有单调性,可以二分答案。
由于答案不可能超过让最快的车(用时最短的车)跑 totalTrips 趟所花费的时间,我们可以将
totalTrips⋅min(time)作为闭区间二分的上界,即让 min(time)这辆车跑 totalTrips 趟,用时 totalTrips⋅min(time),这个时间一定能满足题目要求。
闭区间二分下界:min(time)。
class Solution {
public long minimumTime(int[] time, int totalTrips) {
long minTime = Long.MAX_VALUE;
for (long t : time) {
minTime = Math.min(minTime, t);
}
// 闭区间写法:
// 循环不变量check(left)恒为false
// 循环不变量check(right)恒为true
long left = minTime;
long right = totalTrips * minTime;
while (left <= right) {
long mid = left + (right - left) / 2;
if (check(time, mid, totalTrips)) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return left;
// 左闭右开区间写法:
// long left = minTime;
// long right = totalTrips * minTime + 1;
// while (left < right) {
// long mid = left + (right - left) / 2;
// if (check(time, mid, totalTrips)) {
// right = mid;
// } else {
// left = mid + 1;
// }
// }
// return left;
// 左开右闭区间写法:
// long left = minTime - 1;
// long right = totalTrips * minTime;
// while (left < right) {
// long mid = left + (right - left + 1) / 2;
// if (check(time, mid, totalTrips)) {
// right = mid - 1;
// } else {
// left = mid;
// }
// }
// return left + 1;
// 开区间写法:
// long left = minTime - 1;
// long right = totalTrips * minTime + 1;
// while (left + 1 < right) {
// long mid = left + (right - left) / 2;
// if (check(time, mid, totalTrips)) {
// right = mid;
// } else {
// left = mid;
// }
// }
// return right;
}
private boolean check(int[] time, long x, int totalTrips) {
long sum = 0;
for (long t : time) {
sum += x / t;
if (sum >= totalTrips) {
return true;
}
}
return false;
}
}
复杂度分析
时间复杂度:O(nlogU),其中 n 为 time 的长度,U=totalTrips⋅min(time)。
空间复杂度:O(1)。