一开始想着使用暴力解法,遍历每个加油站,然后以每个加油站为起点跑一圈,最终找到节点,但是这样写在测试一个很长的案例是,超时了,不得不想其他方法去做。
以下是超时的代码:
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
int result = -1;
for(int i = 0; i < gas.size(); i++){
int sum = 0;
sum += gas[i];
int start = i;
for(int j = start; j < gas.size(); j++){
sum -= cost[j];
if(j == gas.size() - 1){
j = -1;
}
if(sum < 0){
break;
}else{
sum += gas[j + 1];
}
if(j == start - 1){
if(sum >= 0){
result = start;
break;
}else{
break;
}
}
}
}
return result;
}
};
使用贪心的思想来做,首先如果总油量减去总消耗大于等于零那么一定可以跑完一圈,说明 各个站点的加油站 剩油量rest[i]相加一定是大于等于零的。i从0开始累加rest[i],和记为curSum,一旦curSum小于零,说明[0, i]区间都不能作为起始位置,因为这个区间选择任何一个位置作为起点,到i这里都会断油,那么起始位置从i+1算起,再从0计算curSum。
那么局部最优:当前累加rest[i]的和curSum一旦小于0,起始位置至少要是i+1,因为从i之前开始一定不行。全局最优:找到可以跑一圈的起始位置。
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
int result = 0;
int sum = 0;
int total = 0;
for(int i = 0; i < gas.size(); i++){
sum += gas[i] - cost[i];
total += gas[i] - cost[i];
if(sum < 0){
result = i + 1;
sum = 0;
}
}
if(total < 0){
return -1;
}
return result;
}
};