从零学算法134

134.加油站

在一条环路上有 n 个加油站,其中第 i 个加油站有汽油 gas[i] 升。

你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。

给定两个整数数组 gas 和 cost ,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1 。如果存在解,则 保证 它是 唯一 的。

示例 1:

输入: gas = [1,2,3,4,5], cost = [3,4,5,1,2]

输出: 3

解释:

从 3 号加油站(索引为 3 处)出发,可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油

开往 4 号加油站,此时油箱有 4 - 1 + 5 = 8 升汽油

开往 0 号加油站,此时油箱有 8 - 2 + 1 = 7 升汽油

开往 1 号加油站,此时油箱有 7 - 3 + 2 = 6 升汽油

开往 2 号加油站,此时油箱有 6 - 4 + 3 = 5 升汽油

开往 3 号加油站,你需要消耗 5 升汽油,正好足够你返回到 3 号加油站。

因此,3 可为起始索引。

示例 2:

输入: gas = [2,3,4], cost = [3,4,3]

输出: -1

解释:

你不能从 0 号或 1 号加油站出发,因为没有足够的汽油可以让你行驶到下一个加油站。

我们从 2 号加油站出发,可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油

开往 0 号加油站,此时油箱有 4 - 3 + 2 = 3 升汽油

开往 1 号加油站,此时油箱有 3 - 3 + 3 = 3 升汽油

你无法返回 2 号加油站,因为返程需要消耗 4 升汽油,但是你的油箱只有 3 升汽油。

因此,无论怎样,你都不可能绕环路行驶一周。

提示:

gas.length == n

cost.length == n

1 <= n <= 10^5^

0 <= gas[i], cost[i] <= 10^4^

  • 暴力解法(超时):最容易想到的,就是暴力解法直接二重循环,从每个点开始往后跑一圈,中途只要出现剩余油量小于 0 的情况就表示无法跑通,否则说明可以行驶一圈。
java 复制代码
  public int canCompleteCircuit(int[] gas, int[] cost) {
      int n = gas.length;
      for(int i = 0; i < n; i++){
      	  // 剩余油量
          int rest = gas[i] - cost[i];
          int index = (i + 1) % n;
          while(rest >= 0 && index != i){
              rest += gas[index] - cost[index];
              index = (index + 1) % n;
          }
          // 剩余油量大于等于 0 且 index 回到了 i
          // 就说明能跑完一圈
          if(rest >= 0 && index == i)return i;
      }
      return -1;
  }
  • 全局贪心:三种情况。
    1. 总油量小于总花费,也就是总的剩余油量 remain 小于 0,那怎么都跑不通一圈
    2. 从 0 开始跑一圈,期间最小剩余油量 min 大于等于 0,说明就没缺过油,跑通了,从 0 开始可行
    3. min 小于 0,说明起点在非 0 点,因为我们是从前往后找到 min,所以我们只要从后往前,直到某个点最终能把负值填平,这个点就是起点。
    4. 比如例子 1,我们在 2 处剩余最少的 -6 的油量;从后往前试图填平,假设在 4 处携带了 -6 的油量,到 3 处剩余 0,说明我们从 3-4 能积攒 6 的油量,继续从 4-0-1-2 此时到 2 处最终会剩余 0 油量而不是 -6,因为 3-4 的 6 油量能把 0-1-2 的 -6 油量填平。并且我们直接从后往前找,也因为 [0,2] 不可行那么这个区间内任何一点到 2 也都不可行。也就是说起点只可能在 [3,4]
java 复制代码
  public int canCompleteCircuit(int[] gas, int[] cost) {
      int n = gas.length;
      // 总剩余油量
      int remain = 0;
      // 从 0 开始跑一圈最少的剩余油量
      int min = Integer.MAX_VALUE;
      for(int i = 0; i < n; i++){
          remain += gas[i] - cost[i];
          if(remain < min)min = remain;
      }
      if(remain < 0) return -1; // 情况 1
      if(min >= 0) return 0; // 情况 2
      // 情况 3
      for(int i = n - 1; i >= 0; i--){
          min += gas[i] - cost[i];
          if(min >= 0)return i;
      }
      return -1;
  }
  • 贪心:既然 [i,j] 的剩余油量小于 0 就表示起点只可能是 [j+1, n],那么我们就用 curRemain 记录当前剩余油量,如果小于 0 就更新起点 start 为当前遍历下标 i + 1,直到跑到终点,只要 remain 不小于 0 那么 start 就为最终答案
java 复制代码
  public int canCompleteCircuit(int[] gas, int[] cost) {
      int curRemain = 0;
      int remain = 0;
      int start = 0;
      for(int i = 0; i < gas.length; i++){
          curRemain += gas[i] - cost[i];
          remain += gas[i] - cost[i];
          // 说明 0,1,2...i 不可能为起点,那么假设从 i+1 刚开始跑
          // 只要 i+1 能跑到终点不出现 remain 小于 0 就表示 i+1 为起点
          if(curRemain < 0){
              start = i + 1;
              curRemain = 0;
          }
      }
      if(remain < 0) return -1; // 情况 1
      return start;
  }
相关推荐
呆呆的猫8 分钟前
【LeetCode】227、基本计算器 II
算法·leetcode·职场和发展
Tisfy10 分钟前
LeetCode 1705.吃苹果的最大数目:贪心(优先队列) - 清晰题解
算法·leetcode·优先队列·贪心·
余额不足1213830 分钟前
C语言基础十六:枚举、c语言中文件的读写操作
linux·c语言·算法
火星机器人life3 小时前
基于ceres优化的3d激光雷达开源算法
算法·3d
虽千万人 吾往矣3 小时前
golang LeetCode 热题 100(动态规划)-更新中
算法·leetcode·动态规划
arnold664 小时前
华为OD E卷(100分)34-转盘寿司
算法·华为od
ZZTC4 小时前
Floyd算法及其扩展应用
算法
lshzdq4 小时前
【机器人】机械臂轨迹和转矩控制对比
人工智能·算法·机器人
2401_858286115 小时前
115.【C语言】数据结构之排序(希尔排序)
c语言·开发语言·数据结构·算法·排序算法
猫猫的小茶馆5 小时前
【数据结构】数据结构整体大纲
linux·数据结构·算法·ubuntu·嵌入式软件