思路如下:
相邻房屋不能偷,即对于一个爷爷结点,要么偷爷爷与孙子,要么只偷儿子,比较两者最大值,利用递归求解。
/**
* 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) {
if(root==null){
return 0;
}
int money=root.val;
if(root.left!=null){
//左节点非空,可以偷其左右结点
money+=rob(root.left.left)+rob(root.left.right);
}
if(root.right!=null){
//右结点非空,可以偷其左右结点
money+=rob(root.right.left)+rob(root.right.right);
}
//比较当前结点能偷到的最大值 与 偷两个儿子结点的最大值之和 的大小
return Math.max(money,rob(root.left)+rob(root.right));
}
}
原题地址:
337. 打家劫舍 III