leetcode 337 打家劫舍 III

2021-10-08 21:00:30 字數 1712 閱讀 3552

在上次打劫完一條街道之後和一圈房屋後,小偷又發現了乙個新的可行竊的地區。這個地區只有乙個入口,我們稱之為「根」。 除了「根」之外,每棟房子有且只有乙個「父「房子與之相連。一番偵察之後,聰明的小偷意識到「這個地方的所有房屋的排列類似於一棵二叉樹」。 如果兩個直接相連的房子在同一天晚上被打劫,房屋將自動報警。

計算在不觸動警報的情況下,小偷一晚能夠盜取的最高金額。

示例 1:

輸入: [3,2,3,null,3,null,1]

3/ \

2 3

\ \

3 1

輸出: 7

解釋: 小偷一晚能夠盜取的最高金額 = 3 + 3 + 1 = 7.

示例 2:

輸入: [3,4,5,1,3,null,1]

3/ \

4 5

/ \ \

1 3 1

輸出: 9

解釋: 小偷一晚能夠盜取的最高金額 = 4 + 5 = 9.

樹形dp,對於每個節點儲存乙個記錄,記錄0表示不用這個節點能得到的最大值,記錄1表示用這個節點能得到的最大值,則轉移方程為:

node[0] = max(node.left) + max(node.right)(如果不用當前的節點,那麼可以選擇左右孩子,也可以不選,所以取左右孩子的最大值之和)

node[1] = node.val + node.left[0] + node.right[0](如果用當前節點,那麼左右孩子用不了,所以是當前節點的值加上不用左右孩子的值)

# definition for a binary tree node.

# class treenode:

# def __init__(self, x):

# self.val = x

# self.left = none

# self.right = none

class

solution

:def

rob(self, root: treenode)

->

int:

ifnot root:

return

0 stack =

[(root,0)

]while stack:

p, stat = stack.pop(

)if stat ==0:

(p,1))

if p.right:

(p.right,0)

)if p.left:

(p.left,0)

)else

: p_left_val = p.left.val if p.left else(0

,0) p_right_val = p.right.val if p.right else(0

,0) p.val =

(max

(p_left_val)

+max

(p_right_val)

, p.val + p_left_val[0]

+ p_right_val[0]

)return

max(root.val)

LeetCode337 打家劫舍

在上次打劫完一條街道之後和一圈房屋後,小偷又發現了乙個新的可行竊的地區。這個地區只有乙個入口,我們稱之為 根 除了 根 之外,每棟房子有且只有乙個 父 房子與之相連。一番偵察之後,聰明的小偷意識到 這個地方的所有房屋的排列類似於一棵二叉樹 如果兩個直接相連的房子在同一天晚上被打劫,房屋將自動報警。計...

leetcode337打家劫舍III

不搶劫該節點,則可以搶劫該節點的左右字節點 注意,node節點不搶不意味著一定要搶它的左右子結點,所以在方法中應該呼叫rob方法,而不是robinclude方法 public int robexclude treenode node return rob node.left rob node.rig...

Leetcode 337 打家劫舍 III

在上次打劫完一條街道之後和一圈房屋後,小偷又發現了乙個新的可行竊的地區。這個地區只有乙個入口,我們稱之為 根 除了 根 之外,每棟房子有且只有乙個 父 房子與之相連。一番偵察之後,聰明的小偷意識到 這個地方的所有房屋的排列類似於一棵二叉樹 如果兩個直接相連的房子在同一天晚上被打劫,房屋將自動報警。計...