环形公交路线上有 n
个站,按次序从 0
到 n - 1
进行编号。我们已知每一对相邻公交站之间的距离,distance[i]
表示编号为 i
的车站和编号为 (i + 1) % n
的车站之间的距离。
环线上的公交车都可以按顺时针和逆时针的方向行驶。
返回乘客从出发点 start
到目的地 destination
之间的最短距离。
示例 1:
输入:distance = [1,2,3,4], start = 0, destination = 1
输出:1
解释:公交站 0 和 1 之间的距离是 1 或 9,最小值是 1。
示例 2:
输入:distance = [1,2,3,4], start = 0, destination = 2
输出:3
解释:公交站 0 和 2 之间的距离是 3 或 7,最小值是 3。
示例 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
C++
class Solution {
public:
int distanceBetweenBusStops(vector<int>& distance, int start, int destination)
{
int left=min(start,destination);
int right=max(start,destination);
int tmp=0;
for(int i=left;i<right;i++)
{
tmp+=distance[i];
}
int sum=accumulate(distance.begin(),distance.end(),0);
int res=sum-tmp;
return min(tmp,res);
}
};
python
class Solution:
def distanceBetweenBusStops(self, distance: List[int], start: int, destination: int) -> int:
left=min(start,destination)
right=max(start,destination)
tmp=0
su=sum(distance)
for i in range(left,right):
tmp+=distance[i]
res=su-tmp
return min(tmp,res)