leetcode337 打家劫舍 III

2021-09-09 05:27:57 字數 1995 閱讀 9331

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

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

示例 1:

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

輸出: 7

示例 2:

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

輸出: 9

遞迴,會超出時間限制:

# 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)

:"""

:type root: treenode

:rtype: int

"""ifnot root:

return

0 root_left = self.rob(root.left.left)

+ self.rob(root.left.right)

if root.left else

0 root_right = self.rob(root.right.left)

+ self.rob(root.right.right)

if root.right else

0return

max(root.val+root_left+root_right, self.rob(root.left)

+self.rob(root.right)

)

用res記錄搶node和不搶node的最大值,減少重複子問題的求解:

# 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)

:"""

:type root: treenode

:rtype: int

"""res = self.helper(root)

return

max(res)

defhelper

(self, node)

: res =[0

,0]# 搶node和不搶node獲得的最大值

ifnot node:

return res

node_left, node_right = self.helper(node.left)

, self.helper(node.right)

res[0]

= node.val + node_left[1]

+ node_right[1]

# 搶node不能搶node.left和node.right

res[1]

=max

(node_left)

+max

(node_right)

# 不搶node,隨便搶不搶node.left和node.right,搶的多就行

return res

LeetCode337 打家劫舍

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

leetcode337打家劫舍III

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

Leetcode 337 打家劫舍 III

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