leetcode 337 打家劫舍 III

2021-10-08 20:54:15 字數 1575 閱讀 3977

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

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

示例 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.

# definition for a binary tree node.

# class treenode(object):

# def __init__(self, x):

# self.val = x

# self.left = none

# self.right = none

class solution(object):

def rob(self, root):

""":type root: treenode

:rtype: int

"""# f表示該節點被選中的最大收益

# g表示該節點沒有被選中的最大收益

f = defaultdict(int)

g = defaultdict(int)

def dfs(node):

if not node:

return

dfs(node.left)

dfs(node.right)

f[node] = node.val + g[node.left] + g[node.right]

g[node] = max(f[node.left],g[node.left]) + max(f[node.right],g[node.right])

dfs(root)

return max(f[root],g[root])

對於每個節點而言,都有兩種狀態,被選中和沒有被選中。我們用兩個雜湊表來儲存每個節點的最大收益,但是需要使用兩張雜湊表,f表示該節點被選中時該點的最大收益,g表示該節點沒有被選中的最大收益。我們可以使用深度優先搜尋來遍歷,使用後序遍歷的方式獲得子節點的資料。

f:當該節點被選中時,那麼子節點必然是不能被選中的,所以f[root] = g[left] + g[right] + 該節點的val

g:當該節點未被選中時,那麼子節點可以被選中也可以沒有選中,所以左兒子的最大貢獻應該是max(f[left], g[left]),同理右兒子最大貢獻是max(f[right], g[right]),最後將其相加就是該節點的最大收益。

LeetCode337 打家劫舍

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

leetcode337打家劫舍III

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

Leetcode 337 打家劫舍 III

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