在一条环路上有 n 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas 和 cost ,如果你可以绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1 。如果存在解,则 保证 它是 唯一 的。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/gas-station
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
求子段和,如果小于0,说明之前的全部不符合,用数组打个标记,从下一个作为开始。最后验证一下是不是可以找到符合的起点。
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
int res=0,f=0,n=gas.size(),cur=0,a[100002];
memset(a,0,sizeof(a));
for(int i=0;;i++)
{
a[i%n]=1;
res+=gas[i%n]-cost[i%n];
if(res<0)
{
res=0;
f=(i+1)%n;
if(a[f]==1)
break;
cur=0;
}
cur++;
if(cur==n)
break;
}
cur=0;res=0;
for(int i=f;;i++)
{
res+=gas[i%n]-cost[i%n];
//cout<<res<<endl;
if(res<0)
{
return -1;
}
cur++;
if(cur==n)
break;
}
return f;
}
};