【算法题】2739. 总行驶距离

插： 前些天发现了一个巨牛的人工智能学习网站，通俗易懂，风趣幽默，忍不住分享一下给大家。点击跳转到网站。
坚持不懈，越努力越幸运，大家一起学习鸭~~~

题目：

卡车有两个油箱。给你两个整数，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

java代码：

class Solution {
    public int distanceTraveled(int mainTank, int additionalTank) {
        int dist = 0;
        while (mainTank > 0) {
            // 消耗油量
            int cost = 0;
            if (mainTank >= 5) {
                mainTank -= 5;
                cost = 5;
                if (additionalTank > 0) {
                    // 每当主油箱使用了5升燃料时，如果副油箱至少有1升燃料，则会将1升燃料从副油箱转移到主油箱。
                    additionalTank -= 1;
                    mainTank += 1;
                }
            } else {
                // 用掉剩下的燃料
                cost = mainTank;
                mainTank = 0;
            }

            // 每耗费1升燃料都可以行驶10km
            dist += cost * 10;
        }

        return dist;
    }
}