LeetCode題目鏈接:https://leetcode-cn.com/problems/house-robber-iii/
該題目在便利蜂面試的時候遇到了,當時完全沒做出來[旺柴]
在上次打劫完一條街道之後和一圈房屋後,小偷又發現了一個新的可行竊的地區。這個地區只有一個入口,我們稱之爲“根”。 除了“根”之外,每棟房子有且只有一個“父“房子與之相連。一番偵察之後,聰明的小偷意識到“這個地方的所有房屋的排列類似於一棵二叉樹”。 如果兩個直接相連的房子在同一天晚上被打劫,房屋將自動報警。
計算在不觸動警報的情況下,小偷一晚能夠盜取的最高金額。
示例 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.
參考了LeetCode的答案,有一種方式很值得學習,
思路:返回一個數組arr,用arr[0]表示不搶root獲得的最大值,用arr[1]表示搶root節點獲得的最大值,那麼最終的值就是在搶root節點和不搶root節點中間找一個最大的。
直接上代碼
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int rob(TreeNode root) {
int[] res = robCore(root);
return Math.max(res[0], res[1]);
}
/**
* 返回數組:arr[0] 表示不搶root獲取的最大zhi
* arr[1] 表示搶root獲取的最大zhi
*/
private int[] robCore(TreeNode node) {
if (node == null) {
return new int[]{0, 0};
}
int[] left = robCore(node.left);
int[] right = robCore(node.right);
//搶 root+下家不能搶
int rob = node.val + left[0] + right[0];
//不搶 下家可搶 可不搶
int not_rob = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
return new int[]{not_rob, rob};
}
}
執行結果: