[LeetCode] 1184. Distance Between Bus Stops 公交站間的距離
A bushasn
stops numbered from0
ton - 1
that forma circle. We know the distance between all pairs of neighboring stops wheredistance[i]
is the distance between the stops numberi
and(i + 1) % n
.
The bus goes along both directionsi.e. clockwise and counterclockwise.
Return the shortest distance between the givenstart
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 <= n<= 10^4
distance.length == n
0 <= start, destination < n
0 <= distance[i] <= 10^4
這道題說是有n個公交站形成了一個環,它們之間的距離用一個數組 distance 表示,其中 distance[i] 表示公交站i和 (i+1)%n
解法一:
class Solution {
public:
int distanceBetweenBusStops(vector<int>& distance, int start, int destination) {
int total = accumulate(distance.begin(), distance.end(), 0), cur = 0, mx = max(start, destination);
for (int i = min(start, destination); i < mx; ++i) {
cur += distance[i];
}
return min(cur, total - cur);
}
};
我們也可以只要一次遍歷就可以完成,因為最終都要是要遍歷所有的站點距離,當這個站點在 [start, destination) 範圍內,就累加到 sum1 中,否則就累加到 sum2 中。不過要要注意的是要確保 start 小於 destination,所以可以開始做個比較,若不滿足則交換二者。最終返回 sum1 和 sum2 中較小者即可,參見程式碼如下:
解法二:
class Solution {
public:
int distanceBetweenBusStops(vector<int>& distance, int start, int destination) {
int sum1 = 0, sum2 = 0, n = distance.size();
if (start > destination) swap(start, destination);
for (int i = 0; i < n; ++i) {
if (i >= start && i < destination) {
sum1 += distance[i];
} else {
sum2 += distance[i];
}
}
return min(sum1, sum2);
}
};
Github 同步地址:
https://github.com/grandyang/leetcode/issues/1184
參考資料:
https://leetcode.com/problems/distance-between-bus-stops/
https://leetcode.com/problems/distance-between-bus-stops/discuss/377568/C%2B%2B-one-pass