lintcode 187. 加油站

在一条环路上有 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;
    }
};
发布了330 篇原创文章 · 获赞 13 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_43981315/article/details/103827958