Question

2739. 总行驶距离

English Version

题目描述

卡车有两个油箱。给你两个整数，mainTank 表示主油箱中的燃料（以升为单位），additionalTank 表示副油箱中的燃料（以升为单位）。

该卡车每耗费 1 升燃料都可以行驶 10 km。每当主油箱使用了 5 升燃料时，如果副油箱至少有 1 升燃料，则会将 1 升燃料从副油箱转移到主油箱。

返回卡车可以行驶的最大距离。

注意：从副油箱向主油箱注入燃料不是连续行为。这一事件会在每消耗 5 升燃料时突然且立即发生。

 

示例 1：

输入：mainTank = 5, additionalTank = 10
输出：60
解释：
在用掉 5 升燃料后，主油箱中燃料还剩下 (5 - 5 + 1) = 1 升，行驶距离为 50km 。
在用掉剩下的 1 升燃料后，没有新的燃料注入到主油箱中，主油箱变为空。
总行驶距离为 60km 。

示例 2：

输入：mainTank = 1, additionalTank = 2
输出：10
解释：
在用掉 1 升燃料后，主油箱变为空。
总行驶距离为 10km 。

 

提示：

• 1 <= mainTank, additionalTank <= 100

解法

方法一：模拟

我们可以模拟卡车的行驶过程，每次消耗 $1$ 升主油箱中的燃料，行驶 $10$ 公里。每当主油箱中的燃料消耗 $5$ 升时，如果副油箱中有燃料，则将副油箱中的 $1$ 升燃料转移到主油箱中。一直模拟到主油箱中的燃料消耗完为止。

时间复杂度 $O(n + m)$，其中 $n$ 和 $m$ 分别是主油箱和副油箱中的燃料数量。空间复杂度 $O(1)$。

class Solution:
  def distanceTraveled(self, mainTank: int, additionalTank: int) -> int:
    ans = cur = 0
    while mainTank:
      cur += 1
      ans += 10
      mainTank -= 1
      if cur % 5 == 0 and additionalTank:
        additionalTank -= 1
        mainTank += 1
    return ans

class Solution {
  public int distanceTraveled(int mainTank, int additionalTank) {
    int ans = 0, cur = 0;
    while (mainTank > 0) {
      cur++;
      ans += 10;
      mainTank--;
      if (cur % 5 == 0 && additionalTank > 0) {
        additionalTank--;
        mainTank++;
      }
    }
    return ans;
  }
}

class Solution {
public:
  int distanceTraveled(int mainTank, int additionalTank) {
    int ans = 0, cur = 0;
    while (mainTank > 0) {
      cur++;
      ans += 10;
      mainTank--;
      if (cur % 5 == 0 && additionalTank > 0) {
        additionalTank--;
        mainTank++;
      }
    }
    return ans;
  }
};

func distanceTraveled(mainTank int, additionalTank int) (ans int) {
  cur := 0
  for mainTank > 0 {
    cur++
    ans += 10
    mainTank--
    if cur%5 == 0 && additionalTank > 0 {
      additionalTank--
      mainTank++
    }
  }
  return
}

impl Solution {
  pub fn distance_traveled(mut main_tank: i32, mut additional_tank: i32) -> i32 {
    let mut cur = 0;
    let mut ans = 0;

    while main_tank > 0 {
      cur += 1;
      main_tank -= 1;
      ans += 10;

      if cur % 5 == 0 && additional_tank > 0 {
        additional_tank -= 1;
        main_tank += 1;
      }
    }

    ans
  }
}