Golang Leetcode 198. House Robber.go

思路

dp解法:
状态方程:
dp[i] = max(dp[i-2]+nums[i],dp[i-1])
到当前的house时,有两种情况:
1.抢了前面一个,那么当前这个不能抢
2.抢了前面的前面的那个,那么当前这个可以抢
所以当前的收益,等于上面两种情况的最大值

code

func rob(nums []int) int {
	if len(nums) == 0 {
		return 0
	}
	if len(nums) == 1 {
		return nums[0]
	}
	dp := make([]int, len(nums))
	dp[0] = nums[0]
	dp[1] = mymax(nums[0], nums[1])
	for i := 2; i < len(nums); i++ {
		dp[i] = mymax(dp[i-2]+nums[i], dp[i-1])
	}
	return dp[len(nums)-1]
}

func mymax(x, y int) int {
	if x > y {
		return x
	}
	return y
}

更多内容请移步我的repo: https://github.com/anakin/golang-leetcode

你可能感兴趣的:(leetcode-golang,golang,leetcode)