【数组】公交站间的距离

题目

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。

提示：

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

解题

方法一：数组

思路

顺时针和逆时针方向都走一遍，取最小。

代码

class Solution {
    public int distanceBetweenBusStops(int[] dists, int start, int dest) {
        if (start > dest) {
            int tmp = dest;
            dest = start;
            start = tmp;
        }
        int dist1 = 0, dist2 = 0;
        for (int i = 0; i < dists.length; i++) {
            if (i >= start && i < dest) dist1 += dists[i];
            else dist2 += dists[i];
        }
        return Math.min(dist1, dist2);
    }
}

class Solution {
public:
    int distanceBetweenBusStops(vector<int>& distance, int start, int destination) {
        if (start > destination) swap(start, destination);
        int dist1 = 0, dist2 = 0;
        for (int i = 0; i < distance.size(); i++) {
            if (i >= start && i < destination) dist1 += distance[i];
            else dist2 += distance[i];
        }
        return min(dist1, dist2);
    }
};