题目描述
环形公交路线上有 n
个站,按次序从 0
到 n - 1
进行编号。我们已知每一对相邻公交站之间的距离,distance[i]
表示编号为 i
的车站和编号为 (i + 1) % n
的车站之间的距离。
环线上的公交车都可以按顺时针和逆时针的方向行驶。
返回乘客从出发点 start
到目的地 destination
之间的最短距离。
样例
输入:distance = [1,2,3,4], start = 0, destination = 1
输出:1
解释:公交站 0 和 1 之间的距离是 1 或 9,最小值是 1。
输入:distance = [1,2,3,4], start = 0, destination = 2
输出:3
解释:公交站 0 和 2 之间的距离是 3 或 7,最小值是 3。
输入:distance = [1,2,3,4], start = 0, destination = 3
输出:4
解释:公交站 0 和 3 之间的距离是 6 或 4,最小值是 4。
限制
1 <= n <= 10^4
distance.length == n
0 <= start, destination < n
0 <= distance[i] <= 10^4
算法
(模拟) $O(n)$
- 求出环形线路的总长度
tot
,然后求出从start
到destination
的一个方向的距离,另一个方向的距离可以直接用tot
减去来得到。
时间复杂度
- 遍历常数环形路线,故时间复杂度为 $O(n)$。
空间复杂度
- 只需常数的额外空间。
C++ 代码
class Solution {
public:
int distanceBetweenBusStops(vector<int>& distance, int start, int destination) {
int n = distance.size();
int tot = 0;
for (int i = 0; i < n; i++)
tot += distance[i];
int ans = 0;
while (start != destination) {
ans += distance[start];
start++;
if (start == n)
start = 0;
}
return min(ans, tot - ans);
}
};