打家劫舍系列 (go)

打家劫舍1

問題描述

你是一個專業的小偷,計劃偷竊沿街的房屋。每間房內都藏有一定的現金,影響你偷竊的唯一制約因素就是相鄰的房屋裝有相互連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警。

給定一個代表每個房屋存放金額的非負整數數組,計算你 不觸動警報裝置的情況下 ,一夜之內能夠偷竊到的最高金額。

思路

動態規劃的思路
1 確定dp
dp[i] 指的是有i間房,此時偷取的最高金額

2 確實狀態變化規則
dp[i]: 有i間房,偷取的最高金額
dp[i-1] : 有i-1間房,。。。。。
dp[i-2] : 有i-2 間房,。。。。。
value [i] :第i間房的金額
因為不能相鄰,dp[i] 和 dp[i-1] 以及dp[i-2]有關
第i間房偷取:dp[i] = dp[i-2] + value[i]
第i間房不偷取: dp[i] = dp [i-1]

dp [i] = max(d[i-1] ,dp[i-2]+value[i]

3 初始化
dp[0] =0
dp[1] = value [0]

4 確定遍歷的順序
從前往后
‘5 舉例
。。。。。。

go語言實現
func rob(nums []int) int {
    res := make([]int,len(nums)+1)
    res[0] = 0 //表示沒房的時候,偷取金額為0
    res[1] = nums[0]
    for i:=2; i<len(res); i++ {
        res[i] = max(res[i-1],res[i-2]+nums[i-1])
    }
    return res[len(nums)]
}
func max(a,b int) int {
    if a>b {
        return a
    } else {
        return b
    }
}

打家劫舍2

問題描述

你是一個專業的小偷,計劃偷竊沿街的房屋,每間房內都藏有一定的現金。這個地方所有的房屋都 圍成一圈 ,這意味著第一個房屋和最后一個房屋是緊挨著的。同時,相鄰的房屋裝有相互連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警 。

給定一個代表每個房屋存放金額的非負整數數組,計算你 在不觸動警報裝置的情況下 ,今晚能夠偷竊到的最高金額。

思路

這個和上面的類似。
不過要考慮到首尾的情況

1 首部不偷取, 即只需 rob( nums[1:])
2 尾部不偷取,rob (nums[:len(nums)-1)
3 都不偷取, rob(nums[1:len(nums)-1)

1和2 包含第三種情況
動態規劃思路和第一個一樣

go語言實現
func rob(nums []int) int {
    if len(nums) == 1 {
        return nums[0]
    }
    res1 , res2 := 0, 0
    res1 = robOne(nums,0,len(nums)-1)
    res2 = robOne(nums,1,len(nums))
    return max(res1,res2)
}
//把代碼抽離出來,避免重復代碼
func robOne(nums []int, start,end int) int {
    res := make([]int,end-start+1)
    res[0] = 0
    res[1] = nums[start]
    start++
    for i:=2; i<len(res); i++ {
        res[i] = max(res[i-1],res[i-2]+nums[start])
        start++
    }
    return res[len(res)-1]
}
func max(a,b int) int {
    if a>b {
        return a
    } else {
        return b
    }
}

打家劫舍3

問題描述

小偷又發現了一個新的可行竊的地區。這個地區只有一個入口,我們稱之為 root 。

除了 root 之外,每棟房子有且只有一個“父“房子與之相連。一番偵察之后,聰明的小偷意識到“這個地方的所有房屋的排列類似于一棵二叉樹”。 如果 兩個直接相連的房子在同一天晚上被打劫 ,房屋將自動報警。

給定二叉樹的 root 。返回 在不觸動警報的情況下 ,小偷能夠盜取的最高金額 。

思路

1 二叉樹遍歷+記憶法
采用中序遍歷:根 + 左 + 右
用 resmap 記錄遍歷過 的樹節點
在遍歷之前,先判斷當前的root(樹)是否以及遍歷過。

   if root== nil {
        return 0
    }
    if _, ok := resmap[root]; ok { 
        return resmap[root]
    } 

2 動態規劃

一個節點有兩個狀態,即偷還是不偷,而這個狀態有左右子節點所決定,
root: value[x,y] #x是當前節點偷的最大金額,y是不偷
root.Left: value[x1,y1]
root.Right: value[x2,y2]

x = root.val + x1 + x2
y = max(x1,y1) + max(x2,y2)
這里需要用到后序遍歷: 左右根

葉子節點往上遍歷,

go語言實現
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
var resmap = make(map[*TreeNode]int)
func rob(root *TreeNode) int {
      res := dan(root)
      return max(res[0],res[1])
}
//動態規劃
func dan(root *TreeNode) []int {
    if root== nil { return []int{0,0} }
    left := dan(root.Left)
    right := dan(root.Right)

    v1 := root.Val + left[0] + right[0]
    v2 := max(left[0],left[1]) + max(right[0],right[1])

    return []int{v2, v1}
}
func max(a,b int) int {
    if a>b {
        return a
    }
    return b
}
//遍歷,記憶
func memory(root *TreeNode) int {
    if root== nil {
        return 0
    }
    if _, ok := resmap[root]; ok { 
        return resmap[root]
    } 
    if root.Left==nil && root.Right == nil {
        return root.Val
    }
    value1 := root.Val
    if root.Left != nil {
        value1 += rob(root.Left.Left) + rob(root.Left.Right)
    }
    if root.Right != nil  {
        value1 += rob(root.Right.Left) + rob(root.Right.Right)
    }

    value2 := rob(root.Left) + rob(root.Right)
    resmap[root] = max(value2,value1)
    return max(value1,value2) 
}
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容

  • 你是一個專業的小偷,計劃偷竊沿街的房屋。每間房內都藏有一定的現金,影響你偷竊的唯一制約因素就是相鄰的房屋裝有相互連...
    想飛的菜鳥閱讀 213評論 0 0
  • 題目 你是一個專業的小偷,計劃偷竊沿街的房屋。每間房內都藏有一定的現金,影響你偷竊的唯一制約因素就是相鄰的房屋裝有...
    golfgang閱讀 384評論 0 1
  • 題目一 你是一個專業的小偷,計劃偷竊沿街的房屋。每間房內都藏有一定的現金,影響你偷竊的唯一制約因素就是相鄰的房屋裝...
    wyof閱讀 199評論 0 0
  • (本文題源LeetCode)LeetCode上有好多題目是有前后關系的,今天我們就來講一下“打家劫舍”三道題的解法...
    奶不加糖閱讀 404評論 0 0
  • 題目 難度:★★☆☆☆類型:數學 你是一個專業的小偷,計劃偷竊沿街的房屋。每間房內都藏有一定的現金,影響你偷竊的唯...
    玖月晴閱讀 981評論 0 0