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

不能连续吃草的牛II

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

知识点:动态规划

由于题目规定,数组成环,故第一个元素和最后一个元素是相邻的,不能同时选择,故我们可以将其分解为两个子问题,对于长度为n的数组来说,我们可以在0到n-1的区间内找到元素和最大值,并且在1到n的区间内找到元素和的最大值,这样就避免了首尾元素相遇的问题,二者中的较大值即为答案。

Java题解如下

import java.util.*;


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

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务