要打劫元素构成了一个环,也就是说在满足不取相邻的数的前提下,第一个和最后一个不能同时取到。于是我们分别去掉第一个和最后一个数,进行两次动态规划,结果最大的就是答案。
class Solution {
public int rob(int[] nums) {
if(nums.length == 1) return nums[0];
return Math.max(func(Arrays.copyOfRange(nums, 1, nums.length)),
func(Arrays.copyOfRange(nums, 0, nums.length-1)));
}
int func(int[] nums){
int n = nums.length;
if(n == 0) return 0;
int dp[][] = new int[2][2];
for(int i = 0; i < n; ++i){
dp[i & 1][0] = Math.max(dp[(i - 1) & 1][0], dp[(i - 1) & 1][1]);
dp[i & 1][1] = dp[(i - 1) & 1][0] + nums[i];
}
return Math.max(dp[(n - 1) & 1][0], dp[(n - 1) & 1][1]);
}
}