213.打家劫舍 II

65 阅读1分钟

题目:
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。

给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,今晚能够偷窃到的最高金额。
算法:
方法一: 动态规划
因为房子连成了环,分类讨论第一个房子和最后一个房子
偷第一个房子,最后一个房子不能偷:可以偷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
}