leetcode之198打家劫舍Golang

题目描述

你是一个专业的小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组，计算你 不触动警报装置的情况下 ，一夜之内能够偷窃到的最高金额。

示例 1：

输入：[1,2,3,1]
输出：4
解释：偷窃 1 号房屋 (金额 = 1) ，然后偷窃 3 号房屋 (金额 = 3)。
     偷窃到的最高金额 = 1 + 3 = 4 。

示例 2：

输入：[2,7,9,3,1]
输出：12
解释：偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9)，接着偷窃 5 号房屋 (金额 = 1)。
     偷窃到的最高金额 = 2 + 9 + 1 = 12 。

提示：

• 0 <= nums.length <= 100
• 0 <= nums[i] <= 400

算法

• 遍历数组，我们假设当前遍历到的数我们作为小偷偷的最后一家，下一家就不偷了，那么我们可以计算出以当前这一家作为最后一家时，我们进行偷盗工作的最大收益
  • 遍历到第一个数，这是第一家，以他家作为结束，那就把他偷了
  • 遍历到第二个数，我们需要斟酌，如果第二家的财产更多就偷第二家，否则偷第一家
  • 遍历到第三个数，我们需要分对第三家是偷还是不偷
    • 偷第三家：那么第二家我们就不能偷，所以我们将第三家偷到的财产加上以第一家作为结束的最大财产，得到在截至第三家偷到所得的累加的总财产
    • 不偷第三家：那么我们截至在第三家偷盗所得的总财产其实就是截至到第二家偷盗所得的总财产。
    • 比较上面两种方式的大小，可以得到截至第三家累加的最大收益
    • 遍历后续数组，每次只需要截至前两家的财产就可以计算出截至到当前家的最大收益
• 返回截至到最后一家的偷盗的收益

代码

func rob(nums []int) int {
	length := len(nums)
	if length == 0 {
		return 0
	}
	if length > 1 {
		// 如果第一家大于第二家，那么就偷第一家，否则就偷第二家
		if nums[1] < nums[0] {
			nums[1] = nums[0]
		}
		if length > 2 {
			for i := 2; i < len(nums); i++ {
				// 偷第i家
				if nums[i]+nums[i-2] >= nums[i-1] {
					nums[i] += nums[i-2]
				} else {
					// 不偷第i家
					nums[i] = nums[i-1]
				}
			}
		}
	}
	return nums[length-1]
}