LeetCode198——打家劫舍

我的LeetCode代码仓:https://github.com/617076674/LeetCode

原题链接:https://leetcode-cn.com/problems/house-robber/description/

题目描述:

LeetCode198——打家劫舍_第1张图片

知识点:动态规划

思路一:动态规划I

状态定义

f(x) -------- 考虑偷取[0, x]范围内的房子。

状态转移

f(x) = max{f(x - 2) + v(x), f(x - 3) + v(x - 1), ... , f(0) + v(2), 0 + v(1), 0 + v(0)}。

时间复杂度是O(n ^ 2),其中n为房屋数量。空间复杂度是O(n)。

JAVA代码:

class Solution {
    public int rob(int[] nums) {
        if(nums.length == 0) {
			return 0;
		}
		//sum[i]:考虑偷取[0, i]范围内的房子
		int[] sum = new int[nums.length];
		sum[0] = nums[0];
		for (int i = 1; i < sum.length; i++) {
			for (int j = 0; j <= i; j++) {
				if(j >= 2) {
					sum[i] = Math.max(sum[i], sum[j - 2] + nums[j]);
				}else {
					sum[i] = Math.max(sum[i], nums[j]);
				}
			}
		}
		return sum[nums.length - 1];
    }
}

LeetCode解题报告:

LeetCode198——打家劫舍_第2张图片

思路二:动态规划II

状态定义

f(x) -------- 考虑偷取[x, n - 1]范围内的房子。

状态转移

f(x) = max{v(x) + f(x + 2), v(x + 1) + f(x + 3), ... , v(n - 3) + f(n - 1), v(n - 2) + 0, v(n - 1) + 0}。

时间复杂度是O(n ^ 2),其中n为房屋数量。空间复杂度是O(n)。

JAVA代码:

class Solution {
    public int rob(int[] nums) {
		if(nums.length == 0) {
			return 0;
		}
		if(nums.length == 1) {
			return nums[0];
		}
		int[] sum = new int[nums.length + 1];
		sum[1] = nums[nums.length - 1];
		sum[2] = Math.max(sum[1], nums[nums.length - 2]);
		for (int i = 3; i <= nums.length; i++) {
			for (int j = 1; j <= i; j++) {
				if(i - j - 1 > 0) {
					sum[i] = Math.max(nums[nums.length - i + j - 1] + sum[i - j - 1], sum[i]);
				}else {
					sum[i] = Math.max(nums[nums.length - i + j - 1], sum[i]);
				}
			}
		}
		return sum[nums.length];
	}
}

LeetCode解题报告:

LeetCode198——打家劫舍_第3张图片

 

你可能感兴趣的:(LeetCode题解)