[leetcode] 1184. Distance Between Bus Stops

Description

A bus has n stops numbered from 0 to n - 1 that form a circle. We know the distance between all pairs of neighboring stops where distance[i] is the distance between the stops number i and (i + 1) % n.

The bus goes along both directions i.e. clockwise and counterclockwise.

Return the shortest distance between the given start and destination stops.

Example 1:

Input: distance = [1,2,3,4], start = 0, destination = 1
Output: 1
Explanation: Distance between 0 and 1 is 1 or 9, minimum is 1.

Example 2:

Input: distance = [1,2,3,4], start = 0, destination = 2
Output: 3
Explanation: Distance between 0 and 2 is 3 or 7, minimum is 3.

Example 3:

Input: distance = [1,2,3,4], start = 0, destination = 3
Output: 4
Explanation: Distance between 0 and 3 is 6 or 4, minimum is 4.

Constraints:

  1. 1 <= n <= 10^4.
  2. distance.length == n.
  3. 0 <= start, destination < n
  4. 0 <= distance[i] <= 10^4

分析

题目的意思是:给定一个数组,求出顺时针遍历和逆时针遍历求和的最小值。我这里用了两次循环来模拟顺时针遍历和逆时针遍历,看了别人的参考发现start和destination可以先比较大小,保持start值小于destination的值然后进行遍历。anyway,我也实现了,时间复杂度O(n),还不错

代码

class Solution:
    def distanceBetweenBusStops(self, distance: List[int], start: int, destination: int) -> int:
        res=0
        n=len(distance)
        t=0
        for i in range(start,n+destination):
            j=i%n
            if(j==destination):
                break
            t+=distance[j]
        res=t
        t=0
        for i in range(destination,n+start):
            j=i%n
            if(j==start):
                break
            t+=distance[j]
            if(t>res):
                return res
        return t

版权声明:本文为w5688414原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。