加油站
题目
在一条环路上有 N 个加油站,其中第 i 个加油站有汽油gas[i],并且从第i个加油站前往第i+1个加油站需要消耗汽油cost[i]。
你有一辆油箱容量无限大的汽车,现在要从某一个加油站出发绕环路一周,一开始油箱为空。
求可环绕环路一周时出发的加油站的编号,若不存在环绕一周的方案,则返回-1。注意事项
数据保证答案唯一。样例
现在有4个加油站,汽油量gas[i]=[1, 1, 3, 1],环路旅行时消耗的汽油量cost[i]=[2, 2, 1, 1]。则出发的加油站的编号为2。
挑战
O(n)时间和O(1)额外空间
题解
贪心法,通过一次遍历找到最后一个盈余汽油为负数的加油站i。如果总盈余为负数则无解,否则解为i+1。
public class Solution {
/**
* @param gas: an array of integers
* @param cost: an array of integers
* @return: an integer
*/
public int canCompleteCircuit(int[] gas, int[] cost) {
int total = 0;
int index = -1;
for(int i = 0, sum = 0; i<gas.length; i++)
{
sum += gas[i] - cost[i];
total += gas[i] - cost[i];
if(sum < 0)
{
index = i;
sum = 0;
}
}
return total < 0 ? -1 : index + 1;
}
}
Last Update 2016.10.28