134.[中] [贪心]加油站问题
题目:给定两个数组,gas[]表示每个加油站的油量,cost[]表示本站到下一站消耗的油量,求是否可以从某一个站点开始,循环走到这个站点,如果可以的话,返回这个站点的标号,不可以返回-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 可为起始索引。
解析:
- 每一个站点都循环一遍,只要能走的回来,那么直接返回,有一个地方走不通,直接开始下一轮循环
- 每个站点循环的过程中,注意当前油量和到下一个站点所需的油量,不够就直接break,开始下一轮循环
- 注意循环数组的更新方式为rear=(rear+1)%length
代码:
private static void start() {
Solution134 solution134=new Solution134();
int[] gas=new int[]{1,2,3,4,5};
int[] cost=new int[]{3,4,5,1,2};
System.out.println(solution134.canCompleteCircuit(gas,cost));
}
public int canCompleteCircuit(int[] gas, int[] cost) {
//从第i个站点出发
for (int i = 0; i < gas.length; i++) {
//记录从i出发的当前油量
int curr=gas[i];
//到下一个站点的消耗
int nextCost=cost[i];
//查看这一圈是否能走的过去,能则直接返回
boolean noPass=false;
//这是从i开始出发,走一圈
for (int j = i; j < i + gas.length; j++) {
//如果当前油量不足以支撑到下一个站点,结束本站点开始的循环
if(curr<nextCost){
noPass=true;
break;
}
//循环的index,如果到达最后一个,那么下一个要这么写,循环数组的标志
int index=(j+1)% gas.length;
//当前油量更新
curr=curr-nextCost+gas[index];
//下一站的消耗更新
nextCost=cost[index];
}
//能走过去就直接返回
if(!noPass){
return i;
}
}
return -1;
}