Leetcode 1184. 公交站间的距离

Leetcode 1184. 公交站间的距离

题目

环形公交路线上有 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。

思路

• 设起点为s， 终点为d
• 如果只按照顺时针来计算， 本体就等同于min(s->d, d->s)

代码 —— golang

func distanceBetweenBusStops(distance []int, start int, destination int) int {
	res, record := math.MaxInt32, 0
	s, d := start, destination

	for s != d {
		record += distance[s]
		s = (s + 1) % len(distance)
	}

	res = int(math.Min(float64(res), float64(record)))

	s, d, record = start, destination, 0

	for s != d {
		record += distance[d]
		d = (d + 1) % len(distance)
	}

	res = int(math.Min(float64(res), float64(record)))

	return res
}