LC 337 打家劫舍 III 樹形DP 遞迴

2021-10-08 20:58:25 字數 1733 閱讀 5220

思路:樹形dpdp

dp+遞迴。

1.考慮考慮某個結點選或不選的的子樹最大值為f(x

),g(

x)

f(x),g(x)

f(x),g

(x)

若選x

xx,則f(x

)=g(

lson

)+g(

rson

)f(x)=g(lson)+g(rson)

f(x)=g

(lso

n)+g

(rso

n)若不選x

xx,則g(x

)=ma

x(f(

lson

),g(

lson

))+m

ax(f

(lso

n),g

(lso

n)

)g(x)=max(f(lson),g(lson))+max(f(lson),g(lson))

g(x)=m

ax(f

(lso

n),g

(lso

n))+

max(

f(ls

on),

g(ls

on))

然後遞迴下去就可以了。

用m ap

mapma

p對映。

```cpp

/** * definition for a binary tree node.

* struct treenode

* };

*/class

solution

introb

(treenode* rt)

};

用結構體優化。

/**

* definition for a binary tree node.

* struct treenode

* };

*/class

solution

; unordered_map,int

>f,g;

node dfs

(treenode * x)

;auto l=

dfs(x-

>left)

;auto r=

dfs(x-

>right)

;int f,g;

f=x-

>val+l.g+r.g;

g=max(l.f,l.g)

+max

(r.f,r.g)

;return;}

introb

(treenode* rt)

};

2.考慮選x

xx後,變為val

+val+

val+

左兒子的左兒子,左兒子的右兒子,右兒子的左兒子,右兒子的右兒子的子樹最大值之和。

不選x

xx,則為左兒子和右兒子的子樹最大值之和也可以進行遞迴。

/**

* definition for a binary tree node.

* struct treenode

* };

*/class

solution

introb

(treenode* rt)

};

337 打家劫舍 III

思路 這是乙個典型的樹形dp問題,對每乙個節點都有兩種選擇 選他或者不選他。選他的話,那麼就不能選擇它的父節點 兄弟節點和子節點 不選這個節點,那麼就可以選擇該節點的左右子樹 選最大的 建立乙個dp表,令dp 0 表示選,dp 1 表示不選。class solution object def rob...

337 打家劫舍 III

方法一 遞迴 超時!class solution 方法二 用雜湊表改進解決重複子問題,超時!class solution return res public int rob treenode root 方法三 用乙個pair分別記錄包含根節點和不包含根節點時的最大值 class solution a...

337 打家劫舍 III

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