https://leetcode-cn.com/problems/house-robber-iii/
使用递归就能解决。区分为两种情况:当前节点用还是不用。
class Solution {
public int rob(TreeNode root) {
if(root == null){
return 0;
}
// rob root
int res1 = root.val;
if(root.left != null){
res1 += rob(root.left.left) + rob(root.left.right);
}
if(root.right != null){
res1 += rob(root.right.left) + rob(root.right.right);
}
//not rob root
int res2 = rob(root.left) + rob(root.right);
return Math.max(res1, res2);
}
}
这种做法会超时,主要原因是存在大量的重复计算。比如:
3
/
2 3
\ \
3 1
在这种情况,抢劫根节点3的时候,需要计算抢不抢叶节点3和1的情况。那么在不抢劫根节点3的时候,直接计算节点2和3,那么此时存在强不强的两种情况,如果不强的时候,那么叶节点3和1就又重复计算了。
如何解决这个问题呢,那么可以用map来进行存储。把一些计算过的节点记录进map里。
class Solution {
public int rob(TreeNode root){
if(root == null){
return 0;
}
HashMap<TreeNode, Integer> map = new HashMap<>();
return help(root,map);
}
public int help(TreeNode root, HashMap<TreeNode, Integer> map) {
if(root == null){
return 0;
}
if(map.containsKey(root)) return map.get(root);
// rob root
int res1 = root.val;
if(root.left != null){
res1 += help(root.left.left,map) + help(root.left.right,map);
}
if(root.right != null){
res1 += help(root.right.left,map) + help(root.right.right,map);
}
//not rob root
int res2 = help(root.left,map) + help(root.right,map);
int res = Math.max(res1, res2);
map.put(root, res);
return res;
}
}
如果map中存在,那么直接取出返回。如果不存在,那么就计算,把计算完的数值记录在map中。
此题也可用动态规划来解决。动态规划的核心就是根据当前状态做出下一步的选择,无需关注是怎么达到当前状态的(也就是无后效性)。经过上面的递归,整个过程是一个调用栈。也就是相当于需要从叶节点开始做选择,进而一步步的向上推进。那么每一步其实都有两个状态,那就是选和不选。所以用一个数组来进行状态的存储。
class Solution {
public int rob(TreeNode root){
int[] res = help(root);
return Math.max(res[0],res[1]);
}
public int[] help(TreeNode root) {
//res[0] : 不选 不rob
//res[1] : 选 rob
int[] res = new int[2];
if(root == null){
return res;
}
int[] left = help(root.left);
int[] right = help(root.right);
res[0] = 0 + Math.max(left[0],left[1]) + Math.max(right[0],right[1]);
res[1] = root.val + left[0] + right[0];
return res;
}
}