加油站问题

来源:互联网 发布:烈焰行动知乎 编辑:程序博客网 时间:2024/04/27 17:35

在一条环路上有 N 个加油站,其中第 i 个加油站有汽油gas[i],并且从第_i_个加油站前往第_i_+1个加油站需要消耗汽油cost[i]

你有一辆油箱容量无限大的汽车,现在要从某一个加油站出发绕环路一周,一开始油箱为空。

求可环绕环路一周时出发的加油站的编号,若不存在环绕一周的方案,则返回-1

样例

现在有4个加油站,汽油量gas[i]=[1, 1, 3, 1],环路旅行时消耗的汽油量cost[i]=[2, 2, 1, 1]。则出发的加油站的编号为2。

数据保证答案唯一。

class Solution {public:    /**     * @param gas: a vector of integers     * @param cost: a vector of integers     * @return: an integer      */    int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {        // write your code here        int n=gas.size();        int flag=0;        int thissum=0;        int allsum=0;        for(int i=0;i<n;i++){            allsum+=gas[i]-cost[i];            thissum+=gas[i]-cost[i];            if(thissum<0){                thissum=0;                flag=i+1;            }        }        return allsum>=0?flag:-1;    }};


0 0