题解 | #不能连续吃草的牛II# java

不能连续吃草的牛II

https://www.nowcoder.com/practice/0b6e9ca056eb4166b4bfd4f7c90b2c61

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param nums int整型一维数组
     * @return int整型
     */
        public int robRange(int[] nums, int start, int end) {
            if (start == end) {
                return nums[start];
            }
            int[] dp = new int[nums.length];
            dp[start] = nums[start];
            dp[start + 1] = Math.max(nums[start], nums[start + 1]);
            for (int i = start + 2; i <= end; i++) {
                dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
            }
            return dp[end];
        }

        public int eatGrass(int[] nums) {
            if (nums.length == 0) {
                return 0;
            }
            if (nums.length == 1) {
                return nums[0];
            }
            int res1 = robRange(nums, 0, nums.length - 2);
            int res2 = robRange(nums, 1, nums.length - 1);
            return Math.max(res1, res2);
        }
    }

编程语言是Java。

该题考察的知识点是动态规划

代码简短的文字解释如下:定义一个dp数组保存中间结果。初始化dp[start]为nums[start],dp[start+1]为max(nums[start], nums[start+1])。然后从start+2开始遍历,dp[i]的值等于前两个状态的最大值加上当前位置的金额,即max(dp[i-2] + nums[i], dp[i-1])。返回dp[end]作为该范围内的最大抢劫金额。

然后在eatGrass函数中,先判断输入数组长度是否为0或1,如果是,则直接返回相应的值。然后分别计算从0到n-2和从1到n-1的两个范围内的最大抢劫金额,取其中较大的一个作为结果返回。

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务