题目描述:
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
示例:
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。
提示:
1 <= nums.length <= 1000 <= nums[i] <= 400
思路:
如果只有一个房屋,就只能选择该房屋,果然两间选择金额较大的进行偷窃。对于要偷窃的房间 k(k>2) 只有两种选择:
1.选择偷窃 选择偷窃该房间 那么偷窃的金额为该房间金额与前k - 2 (不相邻) 房屋最高总金额的和。
2. 不偷窃该房间,偷窃总金额为前 k - 1 间房屋的最高总金额。
那么可得状态转移方程: f(n) (最高总金额) = Math.max(f(n - 2) + nums[n], nums[n - 1])
实现:
package main
import (
"fmt"
)
func main() {
nums := []int{}
fmt.Println(rob(nums)) //12
}
func rob(nums []int) int {
n := len(nums)
if n == 0 {
return 0
}
a, b := 0, nums[0]
for i := 1; i < n; i++ {
a, b = b, max(nums[i]+a, b)
}
return b
}
func max(a, b int) int {
if a > b {
return a
}
return b
}