134. 加油站
在一条环路上有 N 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。
如果你可以绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1。
说明:
如果题目有解,该答案即为唯一答案。
输入数组均为非空数组,且长度相同。
输入数组中的元素均为非负数。
示例 1:
输入:
gas = [1,2,3,4,5]
cost = [3,4,5,1,2]
输出: 3
解释:
从 3 号加油站(索引为 3 处)出发,可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
开往 4 号加油站,此时油箱有 4 - 1 + 5 = 8 升汽油
开往 0 号加油站,此时油箱有 8 - 2 + 1 = 7 升汽油
开往 1 号加油站,此时油箱有 7 - 3 + 2 = 6 升汽油
开往 2 号加油站,此时油箱有 6 - 4 + 3 = 5 升汽油
开往 3 号加油站,你需要消耗 5 升汽油,正好足够你返回到 3 号加油站。
因此,3 可为起始索引。
示例 2:
输入:
gas = [2,3,4]
cost = [3,4,3]
输出: -1
解释:
你不能从 0 号或 1 号加油站出发,因为没有足够的汽油可以让你行驶到下一个加油站。
我们从 2 号加油站出发,可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油
开往 0 号加油站,此时油箱有 4 - 3 + 2 = 3 升汽油
开往 1 号加油站,此时油箱有 3 - 3 + 3 = 3 升汽油
你无法返回 2 号加油站,因为返程需要消耗 4 升汽油,但是你的油箱只有 3 升汽油。
因此,无论怎样,你都不可能绕环路行驶一周。
方法一:多次遍历
直观来看,我们只需遍历gas数组,每一个元素我们都假设是真正的“起点”,再依次进行环状循环验证是否符合真正的起点即可。遍历完全后若还没找到则返回-1即可。
另外注意:
for循环适合模拟从头到尾的遍历,而while循环适合模拟环形遍历,要善于使用while!
int canCompleteCircuit(int* gas, int gasSize, int* cost, int costSize){
if(gasSize==1)//因为下面的循环涉及下一项,所以把一项的情况单拉出来
{
if(gas[0]>=cost[0])
{
return 0;
}
else
return -1;
}
for(int i=0;i<gasSize;i++)
{
if(gas[i]>=cost[i])//首先需要满足这个条件才有可能成为“起点”
{
int sum = gas[i]-cost[i];//sum为我们定义的此时的油量
int j = i+1;
j=j%gasSize;//此为防止j越界的,如果j越界了,就把他变成头部位置,而若不越界,则j大小不变
while(sum>=0&&i!=j)//此时油量大于等于0且在未进行完全环状遍历时
{
sum=sum+gas[j]-cost[j];//更新一下此时油量
j++;//往后一个
j=j%gasSize;//如果越界就通过取余数来将其变成头部
}
if(i==j&&sum>=0)//此为成功的情况,因为只有一解,一旦成功则成功
{
return i;
}
else
{
continue;
}
}
}
return -1;//没成功则必定失败
}
方法二:一次遍历
此方法比前者省去了大量的无效运算。
但使用此方法需要明白两点:
推论1.假如以i为起点,其能到达n位置(n不等于i),则必定真正的起点位置在n之后,不可能为i+1,i+2,i+3等等。
因为:
以i为起点,从i到i+1,其必定积攒了一定油量(也可为0),因为我们已经知道i不能为起点了,所以我们可以看一看起点为i+1时会怎样,而由于以i为起点时到i+1位置还多了i到i+1剩余的油量的情况下都没有成功“跨过”n位置,则少了这些油量的以i+1为起点的情况更不可能“跨过”n位置。
同理我们可以知道i到i+2必然也是积攒了一定油量(也可=0),所以以i+2为起点时其还少了那一部分油量,则更不可能“跨过”n位置。
依次类推,所以i~n都不可能成为真正的起点。
推论2.如1所说,在我们以i为起点到n后,这一部分的计算我们已经做过了,则在之后找到真正的起点后不用再计算这一部分了。因此放到实际数组中即不用进行“环状”操作了。
即:
在n之后的位置依次寻找,若第一次找到了一个m位置符合从m到gas数组末尾都合乎题意,则他就是要找的真正的起点。
因为若此点不是真正的起点,则真正的起点x一定在该点后面,(之所以不在前面是因为若在前面其会比该点更早找到)而如果在该点后面,由于我们前面的推论1可知,从m到x必定会积攒一定的油量(可=0),而该题只有一个解,所以应该是后半部分路程积攒最多油量才为真正的解。因此此想法不可。所以证明推论2也是正确的。
--------注:---------
推论1我们只是证明了i~n不会为真正的起点位置,那么i之前呢?
既然i之前连成为起点的可能都没有(如果他有成为起点的可能,他们自己就会成为另一个“i~n”了),那么他们也必然不可能为真正的起点位置,所以也证明了真正的起点位置在后面。
int canCompleteCircuit(int* gas, int gasSize, int* cost, int costSize){
int total=0,cur=0,get=0;//total为总储存油量,cur为当前油量,get为起点位置,先定为0
for(int i=0;i<gasSize;i++)
{
total=total+gas[i]-cost[i];
cur=cur+gas[i]-cost[i];
if(cur<0)//如推论1,所以归零cur,更改起点位置为i之后
{
get=i+1;
cur=0;
}
}
if(total>=0)//退出循环也可能是无解的情况,所以要保证一定是有解才可
return get;
else
return -1;
}