题解 | #大胃王牛牛#

描述

牧场主有一个牛群的多个牛棚，在环形路线上有 n 个牛棚，其中第 i 个牛棚有 grass[i] 单位的草料。你有一头大胃王牛牛，可以不停吃东西，同时它也很懒惰，不吃东西就不走，它在从第 i 个牛棚走到第 i+1 个牛棚需要消耗草料 cost[i] 单位。现牛牛可以从任一个牛棚出发，初始它没有吃过任何草料。

给定两个整数数组 grass 和 cost ，请判断牛牛应该从哪个牛棚出发，能沿顺序走完环形路线的一个周。若不存在这样的牛棚，则返回 -1。如果存在解答，返回下标最小的。（返回下标从1开始）

示例1

输入：[3, 2, 0, 4],[3, 4, 0, 2]
返回值：3

示例2

输入：[1, 2, 3, 4, 5],[3, 4, 5, 1, 2]
返回值：4

备注：

grass.length == n
cost.length == n
1 <= n <= 10^5
0 <= grass[i], cost[i] <= 10^4

知识点

贪心

解题思路

使用totalGas变量来计算总的油料量，使用currGas变量来记录当前剩余的油料量。

如果当前剩余油料量小于0，说明无法到达下一个牛棚，需要重新选择起点并重置剩余油料量。否则，我们将剩余油料量增加当前牛棚的草料减去消耗的油料。

最后，判断总的油料量是否小于0，如果是，则返回-1表示无法走完一圈；否则返回起点位置加1，即为最终结果。

Java题解

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param grass int整型一维数组
     * @param cost int整型一维数组
     * @return int整型
     */
    public int can_complete_circuit (int[] grass, int[] cost) {
        // write code here
        int n = grass.length;

        int totalGas = 0; // 总的油料量
        int currGas = 0; // 当前剩余油料量
        int start = 0; // 记录起点位置

        for (int i = 0; i < n; i++) {
            totalGas += (grass[i] - cost[i]);

            // 当前剩余油料量小于0，无法到达下一个牛棚
            if (currGas < 0) {
                // 重新开始
                start = i;
                currGas = grass[i] - cost[i];
            } else {
                currGas += (grass[i] - cost[i]);
            }
        }

        // 如果总的油料量小于0，说明无法走完一圈
        if (totalGas < 0) {
            return -1;
        }

        return start + 1;
    }
}