题目:
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,今晚能够偷窃到的最高金额。
算法:
方法一: 动态规划
因为房子连成了环,分类讨论第一个房子和最后一个房子
偷第一个房子,最后一个房子不能偷:可以偷nums的[0, n-1]
不偷第一个房子,能偷最后一个房子:可以偷nums的[1, n-2]
tips:连成环的问题,分类讨论求解,清晰多了
func rob(nums []int) int {
if len(nums) == 1 {
return nums[0]
}
return max(doRob(1, len(nums) - 1, nums), doRob(0, len(nums) - 2, nums))
}
func max(a, b int) int {
if a > b {
return a
}
return b
}
func doRob(start, end int, nums []int) int {
if start > end {
return 0
}
ans := nums[start]
// 抢劫和不抢劫房子的最大价值
notRobVal, robVal := 0, nums[start]
for i := start + 1; i <= end; i ++ {
preRobVal := robVal
robVal = notRobVal + nums[i]
notRobVal = max(preRobVal, notRobVal)
// 最大值肯定出现在抢劫某房子时
if robVal > ans {
ans = robVal
}
}
return ans
}