在一条环路上有 N 个加油站,其中第 i 个加油站有汽油gas[i],并且从第_i_个加油站前往第_i_+1个加油站需要消耗汽油cost[i]。
你有一辆油箱容量无限大的汽车,现在要从某一个加油站出发绕环路一周,一开始油箱为空。
求可环绕环路一周时出发的加油站的编号,若不存在环绕一周的方案,则返回-1。
样例
样例 1:
输入:gas[i]=[1,1,3,1],cost[i]=[2,2,1,1]
输出:2
样例 2:
输入:gas[i]=[1,1,3,1],cost[i]=[2,2,10,1]
输出:-1
挑战
O(n)时间和O(1)额外空间
注意事项
数据保证答案唯一。
贪心
题目说数据保证答案唯一,也就是要么有且仅有一个这样的加油站,要么没有,先看没有的情况,没有的话则说明,走一圈下来,加油站能加的油比消耗的油要少,这种情况下就没有答案。
那么遍历两个数组,先假设这个加油站是0号,然后随着遍历计算累计值,一旦累计值小于0,说明这之前的加油站都是不满足条件的,把开始的设置为遍历的下一个,然后继续这个过程,直到遍历结束。同时计算所有的gas和cost的和,如果sum(gas)<sum(cost),则不可能存在这样的一个加油站。
class Solution {
public:
/**
* @param gas: An array of integers
* @param cost: An array of integers
* @return: An integer
*/
int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
// write your code here
int sum=0;
int totalgas=0;
int totalcost=0;
int start=0;
for (int i = 0; i < gas.size(); i++) {
/* code */
sum=sum+gas[i]-cost[i];
totalgas+=gas[i];
totalcost+=cost[i];
if(sum<0)
{
sum=0;
start=i+1;
}
}
return totalgas>=totalcost?start:-1;
}
};