仅为个人记录复盘学习历程,解题思路来自代码随想录
代码随想录刷题笔记总结网址:
代码随想录
134. 加油站
在一条环路上有 N 个加油站,其中第 i 个加油站有汽油 gas[i] 升。你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。如果你可以绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1。
关键思路:
如果gas[i]-cos[i]的i从0开始到结束到的累加和小于0,则说明油是不够的,返回-1;
如果gas[i]-cos[i]的i从0开始到结束到的累加和大于等于0,则说明油是够的,
使用贪心算法,对于找到能从该加油站回到该加油站的起始加油站,
假设rest[i]=gas[i]-cos[i],
局部最优:i从0开始,求rest[i]的累加和,如果累加和小于0,则起始节点不在[0,i]的区间,从i+
节点开始寻找起始加油站。(从0节点开始,假设到i节点累加和小于0,则在0-i中选的节点在i这个位置一定油不够)
全局最优:找到能从该加油站回到该加油站的起始加油站。
代码大致如下:
public int canCompleteCircuit(int[] gas, int[] cost) {
//初始化变量
int curSum=0;
int totalSum=0;
int startIndex=0;
for(int i=0;i<gas.length;i++){
curSum+=gas[i]-cost[i];
totalSum+=gas[i]-cost[i];
if(curSum<0){
startIndex=i+1;
curSum=0;
}
}
if(totalSum<0)return -1;
return startIndex;
}