算法-贪心篇08-加油站

加油站

力扣题目链接

题目描述

在一条环路上有 n 个加油站，其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车，从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发，开始时油箱为空。
给定两个整数数组 gas 和 cost ，如果你可以按顺序绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1 。如果存在解，则 保证 它是 唯一 的。

解题思路

先从0开始跑一圈，计算当前剩下多少油，并把最小值记录下来。
如果最后油是负数，说明总加油量小于总消耗量，肯定不可能跑一圈；
如果最后，最小油量不为负数，说明路上没有一次缺油，0就是答案；
再尝试从后向前跑，找到一个站，这个站的剩余油耗可以填补最小油量的损耗，说明从这里开始，可以完成一圈。

题解

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int leftGas = 0;
        int min = INT_MAX;
        for(int i = 0; i < gas.size(); i++){
            int rest = gas[i] - cost[i];
            leftGas += rest;
            if(leftGas < min){
                min = leftGas;
            }
        }
        if(leftGas < 0) return -1;
        if(min >= 0) return 0;

        for(int i = gas.size() - 1; i >= 0; i--){
            int rest = gas[i] - cost[i];
            min += rest;
            if(min >= 0){
                return i;
            }
        }

        return -1;
    }
};