题解 | #不能连续吃草的牛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]); } }