本文参考代码随想录的解法,增加了一些理解~
源码来自 134. 加油站
暴力解法:
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
// 暴力解法
for(int i=0;i<cost.size();i++){
int rest=gas[i]-cost[i];
// 在找下一步时,由于需要搭建一个环状,需要用一个取余算法找到净的下一步
int index=(i+1)%cost.size();
// 以当前i为起点,走一圈,控制其终点不能超过i
while(rest>0&&index!=i){
// 当前剩余油量为正,就走下一步
rest+=gas[index]-cost[index];
// 更新index
index=(index+1)%cost.size();
}
// 如果以i为起点走一圈剩余油量还有,就返回这个起始位置-即加油站编号
if(rest>=0&&index==i)return i;
}
return -1;
}
};
结果直接超出了时间限制,当输入的gas和cost超级大的时候会占用很大的内存
全局思路解法:
看经过每个站点都会剩下多少油,如果都是正数那就从哪开始都可以了,也可以直接看总油量减去总消耗,如果大于等于零就一定能跑完一圈
如果有负数,就看从哪开始能把这个负数填平,找到所有能填平的起始节点。当然直接看比较麻烦,可以看如果从0开始累加到i之后curSum小于0,就说明从0到i之间的任何一个位置都不能作为起始位置,只能从i+1开始重新计算curSum。
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
int curSum = 0;
int totalSum = 0;
int start = 0;
for (int i = 0; i < gas.size(); i++) {
curSum += gas[i] - cost[i];
//curSUM有更新,但totalSum始终是在累加的,最后求出来整体的剩余油量,不管从哪开始
totalSum += gas[i] - cost[i];
if (curSum < 0) { // 当前累加rest[i]和 curSum一旦小于0
start = i + 1; // 起始位置更新为i+1
curSum = 0; // curSum从0开始
// 然后会继续往后加,直到curSum又遇到0或者i走到底
//如果走到底,curSum还是正数,就通过totalSum来判断到底有没有符合要求的起点
}
}
if (totalSum < 0) return -1; // 说明怎么走都不可能跑一圈了,不存在符合要求的起点
//如果totalSum大于等于0,说明存在这样的起点,难就能从上面循环中得到这个起点,即为所求
return start;
}
};
javascript版本:
var canCompleteCircuit = function(gas, cost) {
const gasLen = gas.length
let start = 0
let curSum = 0
let totalSum = 0
for(let i = 0; i < gasLen; i++) {
curSum += gas[i] - cost[i]
totalSum += gas[i] - cost[i]
if(curSum < 0) {
curSum = 0
start = i + 1
}
}
if(totalSum < 0) return -1
return start
};