DAY39-动态规划-打家劫舍

LeetCode198.打家劫舍

    public int rob(int[] nums) {
    	int[] dp = new int[nums.length];
    	if(nums.length==1)return nums[0];
    	if(nums.length==2)return Math.max(nums[0], nums[1]);
    	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-1], dp[i-2]+nums[i]);
    	}
    	
    	return dp[nums.length-1];
    }

LeetCode213.打家劫舍II

    public int rob(int[] nums) {
    	if(nums.length==1)return nums[0];
    	if(nums.length==2)return Math.max(nums[0], nums[1]);
    	int[] dp = new int[nums.length];
    	int[] dp2 = new int[nums.length];
    	
    	dp[0]=nums[0];
    	dp[1]=Math.max(nums[0], nums[1]);
    	
    	for(int i=2;i<nums.length-1;i++) {
    		dp[i]=Math.max(dp[i-1], dp[i-2]+nums[i]);
    	}
    	
    	dp2[1]=nums[1];
    	dp2[2]=Math.max(nums[1], nums[2]);
    	
    	for(int i=3;i<nums.length;i++) {
    		dp2[i]=Math.max(dp2[i-1], dp2[i-2]+nums[i]);
    	}

    	return Math.max(dp[nums.length-2], dp2[nums.length-1]);
    }

LeetCode337.打家劫舍III

    public int rob(TreeNode root) {
    	int[] dp = dfs(root);
    	return Math.max(dp[0], dp[1]);
    }
    
    public int[] dfs(TreeNode root) {
    	if(root==null)return new int[] {0,0};
    	int[] left = dfs(root.left);
    	int[] right = dfs(root.right);
    	//如果放根节点
    	int put = root.val+left[0]+right[0];
    	int notPut = Math.max(left[0], left[1])+Math.max(right[0], right[1]);
    	//返回
    	return new int[] {notPut,put};
    }