LeetCode 1184. 公交站间的距离(python)

题目链接

题目描述:

环形公交路线上有 n 个站,按次序从 0 到 n - 1 进行编号。我们已知每一对相邻公交站之间的距离,distance[i] 表示编号为 i 的车站和编号为 (i + 1) % n 的车站之间的距离。

环线上的公交车都可以按顺时针和逆时针的方向行驶。

返回乘客从出发点 start 到目的地 destination 之间的最短距离。

示例 1:
LeetCode 1184. 公交站间的距离(python)_第1张图片

输入:distance = [1,2,3,4], start = 0, destination = 1
输出:1
解释:公交站 0 和 1 之间的距离是 1 或 9,最小值是 1。

示例 2:

LeetCode 1184. 公交站间的距离(python)_第2张图片

输入:distance = [1,2,3,4], start = 0, destination = 2
输出:3
解释:公交站 0 和 2 之间的距离是 3 或 7,最小值是 3。

示例 3:

LeetCode 1184. 公交站间的距离(python)_第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

解题思路:

由于可以顺时针,也可以逆时针行驶,所以距离等于start到destination之间的距离,或者总的距离减去上述距离,可以利用切片方法求距离的和。
由于可能出现destination比start小的情况,所以将两数之中小的赋给start,大的赋给destination

class Solution:
    def distanceBetweenBusStops(self, distance: List[int], start: int, destination: int) -> int:
        start,destination=min(start,destination),max(start,destination)
        return min(sum(distance[start:destination]),sum(distance)-sum(distance[start:destination]))

LeetCode 1184. 公交站间的距离(python)_第4张图片

你可能感兴趣的:(leetcode,LeetCode)