加油站(力扣134)
既然每一个加油站都有对应的加油量和耗油量,我们不妨计算一下每个加油站的汽油净增量。如果每个加油站净增量之和不为负数,则说明一定可以找到唯一的起始点。那我们该如何找到这个起始点呢?我们设置最开始的起点为第0个加油站,接着通过for循环往后遍历每一个加油站,同时将每个加油站的净增量逐个累加。在这个过程中我们运用贪心思想:当汽油净增量之和为负数时,我们就可以将起始点更新到当前加油站的下一位。大家可以结合我下面的代码及详细注释理解此题。
代码及详细注释如下:
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
vector<int> sum(gas.size() + 1);
//计算每个加油站的油量净增多少
for(int i = 0;i < gas.size();i++){
sum[i] = gas[i] - cost[i];
}
int total = 0;
for(int i = 0;i < sum.size();i++){
total += sum[i];
}
//用全局视先判断问题是否存在解
if(total < 0) return -1;
int Cursum = 0;
int start = 0;
for(int i = 0;i < sum.size();i++){
Cursum += sum[i];
//当前汽油为负数时,就要将下一位设置为新的start
if(Cursum < 0){
//记得清0当前汽油量
Cursum = 0;
start = i + 1;
continue;
}
}
return start;
}
};