一、打家劫舍(LeetCode198)
class Solution {
public int rob(int[] nums) {
if(nums == null || nums.length == 0) return 0;
if(nums.length == 1) return nums[0];
int[] dp = new int[nums.length];
dp[0] = nums[0];
dp[1] = Math.max(nums[0],nums[1]);
for(int i = 2;i < nums.length;i++){
dp[i] = Math.max(dp[i-2]+nums[i],dp[i-1]);
}
return dp[nums.length - 1];
}
}
二、打家劫舍II(LeetCode213)
class Solution {
public int rob(int[] nums) {
if(nums == null || nums.length == 0) return 0;
if(nums.length == 1) return nums[0];
int result1 = robRange(nums, 0, nums.length - 2);
int result2 = robRange(nums, 1, nums.length - 1);
return Math.max(result1, result2);
}
public int robRange(int[] nums,int start,int end){
if (end == start) 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];
}
}
三、打家劫舍III(LeetCode337)
class Solution {
public int rob(TreeNode root) {
if(root.left == null && root.right == null) return root.val;
int[] res = robTree(root);
return Math.max(res[0], res[1]);
}
int[] robTree(TreeNode root) {
int res[] = new int[2];
if (root == null)
return res;
int[] left = robTree(root.left);
int[] right = robTree(root.right);
res[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
res[1] = root.val + left[0] + right[0];
return res;
}
}