加油站-LintCode

来源:互联网 发布:淘宝女鞋 编辑:程序博客网 时间:2024/04/29 23:15

在一条环路上有 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)额外空间

#ifndef C187_H#define C187_H#include<iostream>#include<vector>using namespace std;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        if (gas.size() != cost.size() || gas.size() == 0 || cost.size() == 0)            return -1;        int sumGas = 0, sumCost = 0;        int pos = 0;        int cur = 0;        for (int i = 0; i < gas.size(); ++i)        {            cur += gas[i];            cur -= cost[i];            sumGas += gas[i];            sumCost += cost[i];            if (cur < 0)            {                pos = i + 1;                cur = 0;            }        }        return sumGas>=sumCost?pos:-1;    }};#endif
原创粉丝点击