在一条环路上有 N 个加油站,其中第 i 个加油站有汽油gas[i]
,并且从第_i_个加油站前往第_i_+1个加油站需要消耗汽油cost[i]
。
你有一辆油箱容量无限大的汽车,现在要从某一个加油站出发绕环路一周,一开始油箱为空。
求可环绕环路一周时出发的加油站的编号,若不存在环绕一周的方案,则返回-1
。
Yes
样例
现在有4个加油站,汽油量gas[i]=[1, 1, 3, 1]
,环路旅行时消耗的汽油量cost[i]=[2, 2, 1, 1]
。则出发的加油站的编号为2。
注意
数据保证答案唯一。
挑战
标签
Expand
O(n)时间和O(1)额外空间
分析:只要总汽油量要大于总的消耗量,那么肯定是有解的,可以从头遍历起,什么时候汽油量小于消耗量了,就假设从下一个点重新开始。
#include <iostream>
#include <vector>
using namespace std;
int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
int total = 0;
int j = -1;
for (int i = 0, sum = 0; i < gas.size(); ++i) {
sum += gas[i] - cost[i];
total += gas[i] - cost[i];
if (sum < 0) {
j = i;//不断更新,最后一次 sum<0 的下标j,
sum = 0;//清空sum重新从i+1开始判断。
}
}//只要总汽油量要大于总的消耗量,那么肯定是有解的,不用构成环来判断。
return total >= 0 ? j + 1 : -1;
}
void main(){
int gasArr[]={1, 1, 3, 1};
int costArr[]={2, 2, 1, 1};
int gasLen=sizeof(gasArr)/sizeof(int);
vector<int> gas(gasArr,gasArr+gasLen);
int costLen=sizeof(costArr)/sizeof(int);
vector<int> cost(costArr,costArr+costLen);
cout<<"加油站的编号:"<<canCompleteCircuit(gas,cost) <<endl;
system("pause");
}