28 lines
1.0 KiB
C++
28 lines
1.0 KiB
C++
|
#include "s0134_gas_station.hpp"
|
|||
|
|
|||
|
// 我们假设 rest[i] = gas[i] - cost[i] 等于当前加油站所能积累的油量
|
|||
|
// 情况一:
|
|||
|
// 如果 totalSum = rest[0] + rest[1] + ... + rest[n-1] < 0 则说明所有加油站的油加起来都不够跑一圈的
|
|||
|
// 情况二:
|
|||
|
// 我们假设 totalSum >= 0 也就是所有加油站的油加起来能跑一圈
|
|||
|
// 当前加油站索引为 i ,起始节点为 startIndex
|
|||
|
// 那么如果 curSum = rest[startIndex] + rest[startIndex+1] + ... + rest[i] < 0 则说明剩下的节点加起来一定大于零
|
|||
|
// 也就是 i 之前出现了多少负数,之后就应该出现多少正数
|
|||
|
// 这时把 startIndex 设为 i+1 ,curSum = 0 继续迭代
|
|||
|
int S0134::canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
|
|||
|
int curSum{0};
|
|||
|
int totalSum{0};
|
|||
|
int startIndex{0};
|
|||
|
int len = gas.size();
|
|||
|
for (int i{0}; i < len; ++i) {
|
|||
|
curSum += gas[i] - cost[i];
|
|||
|
totalSum += gas[i] - cost[i];
|
|||
|
if (curSum < 0) {
|
|||
|
startIndex = i + 1;
|
|||
|
curSum = 0;
|
|||
|
}
|
|||
|
}
|
|||
|
if (totalSum < 0) return -1;
|
|||
|
return startIndex;
|
|||
|
}
|