198.M 打家劫舍

Problem: 198. 打家劫舍

思路

动态规划。

状态转移方程：

dp[i] = max(dp[i-2]+nums[i], dp[i-1])

dp[i]表示前i个房屋的最大金额

Code

代码1：

func rob(nums []int) int {
	if len(nums) == 0 {
		return 0
	}

	if len(nums) == 1 {
		return nums[0]
	}

	dp := make([]int, len(nums))
	dp[0] = nums[0]
	dp[1] = max(nums[0], nums[1])

	for i := 2; i < len(nums); i++ {
		dp[i] = max(dp[i-2]+nums[i], dp[i-1])
	}

	return dp[len(nums)-1]
}

代码2：空间优化后的版本

func rob(nums []int) int {
	if len(nums) == 0 {
		return 0
	}

	if len(nums) == 1 {
		return nums[0]
	}

	pre2 := nums[0]
	pre1 := max(nums[0], nums[1])
	for i := 2; i < len(nums); i++ {
		cur := max(pre2+nums[i], pre1)
		pre2 = pre1
		pre1 = cur
	}

	return pre1
}