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

不能连续吃草的牛II

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

一、知识点:

数组

二、文字分析:

将这个问题分解成两个独立的子问题:

  • 问题1:不吃最后一块草,即计算nums[0]到nums[n-2]范围内的最高饱腹感。
  • 问题2:不吃第一块草,即计算nums[1]到nums[n-1]范围内的最高饱腹感。

时间复杂度为O(n),

空间复杂度为O(n)。

三、编程语言:

java

四、正确代码:

import java.util.*;


public class Solution {
    public int eatGrass(int[] nums) {
        int n = nums.length;
        if (n == 0) return 0;
        if (n == 1) return nums[0];

        // 计算不吃最后一块草的最高饱腹感
        int max1 = dp(nums, 0, n - 2);

        // 计算不吃第一块草的最高饱腹感
        int max2 = dp(nums, 1, n - 1);

        // 返回两个子问题结果的较大值
        return Math.max(max1, max2);
    }

    // 动态规划函数,计算nums[start]到nums[end]范围内的最高饱腹感
    private int dp(int[] nums, int start, int end) {
        int n = end - start + 1;
        if (n == 1) return nums[start];

        int[] dp = new int[n];
        dp[0] = nums[start];
        dp[1] = Math.max(nums[start], nums[start + 1]);

        for (int i = 2; i < n; i++) {
            dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[start + i]);
        }

        return dp[n - 1];
    }
}

全部评论

相关推荐

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