leetcode 198.打家劫舍1 golang实现

最新推荐文章于 2021-02-09 20:55:11 发布

tzh_linux

最新推荐文章于 2021-02-09 20:55:11 发布

阅读量172

点赞数

CC 4.0 BY-SA版权

分类专栏：算法数据结构 go leetcode/剑指offer 文章标签： golang leetcode 算法

本文链接：https://blog.youkuaiyun.com/tzh_linux/article/details/103283938

算法数据结构同时被 3 个专栏收录

76 篇文章

订阅专栏

76 篇文章

订阅专栏

leetcode/剑指offer

75 篇文章

订阅专栏

本文探讨了一个有趣的问题：如何在一个整数数组中选择元素，使得相邻元素不可同时选择，目标是使所选元素之和最大。通过动态规划的方法，我们实现了两个解决方案，其中一个优化了空间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述

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

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

示例 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 。

思路

dp[i] 表示抢第i家的时候所得金额
当抢劫第i家时， 判断抢第i- 1 和第i-2家+ 今天要抢的房屋的金额
dp[i] = max(dp[i-2] + curNum, dp[i - 1])

实现

func Rob1(nums []int) int {
    var dp = make([]int, 0)
    if len(nums) == 0{
        return 0
    }

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

    dp = append(dp, 0)
    dp = append(dp, nums[0])

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

    return dp[len(nums)]
}

func MyMax(a, b int)int{
    if a >= b{
        return a
    }else{
        return b
    }
}

// 优化一下 降低空间复杂度
func Rob2(nums []int) int {
	if len(nums) == 0{
		return 0
	}

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

	pre := 0
	cur := nums[0]

	for i := 2; i <= len(nums); i++{
		tmpCur := cur
		cur= MyMax(pre + nums[i - 1], cur)
		pre = tmpCur
		
	}


	return cur
}