【二分查找】leetcode 1870. 准时到达的列车最小时速

1870. 准时到达的列车最小时速

题目描述

给你一个浮点数 hour ，表示你到达办公室可用的总通勤时间。要到达办公室，你必须按给定次序乘坐 n 趟列车。另给你一个长度为 n 的整数数组 dist ，其中 $dist[i]$ 表示第 i 趟列车的行驶距离（单位是千米）。

每趟列车均只能在整点发车，所以你可能需要在两趟列车之间等待一段时间。

例如，第 1 趟列车需要 1.5 小时，那你必须再等待 0.5 小时，搭乘在第 2 小时发车的第 2 趟列车。
返回能满足你准时到达办公室所要求全部列车的 最小正整数 时速（单位：千米每小时），如果无法准时到达，则返回 -1 。

生成的测试用例保证答案不超过 $10^7$ ，且 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 小时发车。

提示

• $n==dist.length$
• $1<=n<=10^5$
• $1<=dist[i]<=10^5$
• $1<=hour<=10^9$
• $hours中，小数点后最多存在两位数字$

方法：二分查找

解题思路

由题意得，当时速越高时，所用时越少；当时速越低时，所用时越多。因此，本题具有单调性，可以利用二分答案来解题。

题目要我们求准时到达的列车最小时速，可以转化理解为在不超过 $hour$ 小时，确保准时到达的情况下，最低的时速为多少。

二分答案的左边界为 $1$，右边界为 $10^7$，那如何判断当前 $speed$ 是否符合条件呢？
遍历数组，每一个$dist[i]$ 表示到达一个站点的路程，则对达到每一站计算所需要的时间为
$$\lceil \frac{dist[i]+speed-1}{speeed}\rceil (上取整)，$$
而达到最后一站所需的时间不用取整，用 $1.0\ast dist[i]/speed$。

• 如果总耗时 $hours<=hour$，则将区间缩小为 $[l,mid]$；
• 如果总耗时 $hours>hour$，则将区间缩小为 $[mid+1,r]$；

直到 $l=r$ 结束循环。

最后再判断一下，当 $speed=l$ 时，$hours是否<=hour$，如果不满足，则返回 $-1$。

代码

class Solution {
public:
    double cal(vector<int>& dist, int speed) {
        double hours = 0;
        int n = dist.size();
        for(int i = 0; i < n; i++)
            if(i < n - 1)
                hours += (dist[i] + speed - 1) / speed;
            else
                hours += 1.0 * dist[i] / speed;
        return hours;
    }
    int minSpeedOnTime(vector<int>& dist, double hour) {
        int l = 1, r = 1e7, mid;
        while(l < r)
        {
            mid = (l + r) >> 1;
            double hours = cal(dist, mid);
            if(hours <= hour)
                r = mid;
            else
                l = mid + 1;
        }
        double res = cal(dist, l);
        return res <= hour ? l : -1;
    }
};

复杂度分析

• 时间复杂度：$O(nlogC)$。$C$ 为二分查找的上下限之差。二分查找的时间复杂度为 $O(logC)$，每一轮二分查找都要遍历数组需要 $O(n)$ 的时间复杂度。因此，整体时间复杂度为 $O(nlogC)$。
• 空间复杂度：$O(1)$。