There are N gas stations along a circular route, where the amount of gas at station i is gas[i].
You have a car with an unlimited gas tank and it costscost[i] of gas to travel from stationi to its next station (i+1). You begin the journey with an empty tank at one of the gas stations.
Return the starting gas station's index if you can travel around the circuit once, otherwise return -1.
Note:
The solution is guaranteed to be unique.
这道题给出一组加油站储存的油量和加油站之间的油耗,问能否从一个加油站出发依次走完所有加油站回到原点,题目难度为Medium。
对于任意加油站m,如果gas[m]<cost[m],则不能从m出发,直接查看下一个加油站。如果gas[m]>=cost[m],则可以从m出发,每经过一个加油站统计剩下的油量,如果剩下油量为负,表明从m出发已不能到达此加油站,将此加油站记为n。从m到n之间的任意加油站出发也不能到达n,为什么呢?取m和n之间的任一加油站p,从m到p剩下的油量肯定不为负,因而从p到n剩下的油量比从m到n剩下的油量小(或相等),也必定为负,所以从m和n之间任意加油站不能到n。这样我们接着检查n的下一个加油站直至找到答案。具体代码:
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
if(gas.empty()) return -1;
int sz = gas.size();
int idx = 0, pos = 0;
int left = 0;
while(idx < sz) {
if(gas[idx] < cost[idx]) {
idx++;
}
else {
while(left >= 0) {
if(pos == idx + sz) return idx;
left += (gas[pos%sz]-cost[pos%sz]);
pos++;
}
left = 0;
idx = pos;
}
pos = idx;
}
return -1;
}
};
这种方法时间复杂度为O(n),所有加油站遍历次数不超过两次。查看别人的代码有仅遍历一次的方法,先上代码:
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
if(gas.empty()) return -1;
int idx = 0;
int left = 0, total = 0;
for(int i=0; i<gas.size(); i++) {
left += gas[i]-cost[i];
if(left < 0) {
total += left;
left = 0;
idx = i + 1;
}
}
total += left;
return (total>=0) ? idx : -1;
}
};
为什么遍历一次最后记录的起点就是满足条件的呢?这里基于两点:第一,题目限定结果唯一或无解;第二,如果gas[0]+gas[1]+...+gas[n-1] > cost[0]+cost[1]+...+cost[n-1],则必定存在一个加油站,从此加油站开始能够顺利走完一圈回到原点。
第一点是题目限定的,不再说明。第二点真的成立吗?试着举反例很久没有想明白,所以查看了别人的证明,具体如下:
- 只有一个加油站:结论成立,不再说明。
- 两个加油站a和b:如果gas[a]>=cost[a]、gas[b]>=cost[b],则从任意加油站开始都可以走完(题目限定结果唯一,不包括此情况也无所谓)。如果gas[a]<cost[a],则gas[b]>cost[b],这样能够从b出发顺利到达a,同时由于gas[a]+gas[b]>cost[a]+cost[b],所以到达a后仍能够顺利回到b;同理gas[b]<cost[b]结论一样成立。
-
三个加油站a、b和c:如果gas[a]>cost[a]、gas[b]>cost[b]、gas[c]>cost[c],和两个加油站的情况相同。如果gas[a]<cost[a],此时又分两种情况:
-
gas[b]>=cost[b],能够从b出发顺利到达c,这样可以把b和c合并为一个加油站b',就把问题化解为两个加油站a和b'的情况,已证。
-
gas[b]<cost[b],则gas[c]>cost[c],能够从c出发顺利到达a,这样把c和a合并为一个加油站c',同上把问题化解为两个加油站c'和b的情况,已证。
其他的情况也能够按照上面的方法化解为两个加油站的情况。
-
这样能够把任意加油站的情况按照上述方法逐次转化为两个加油站,结论成立。
由上面提到的两点能够确定:如果最后剩下的油量为负则无解,如果为正则最后记录的起点即是结果。