Leetcode337. 打家劫舍 III（C语言）

Leetcode337. 打家劫舍 III（C语言）

数据结构-树：算法与数据结构参考

题目：
在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。例:
输入: [3,4,5,1,3,null,1]
输出: 9

思路：
法1：动态规划¹（dp）+深度优先搜索（dfs）/递归
若偷了根节点，就不能偷它的子节点，所以分两种情况
不能直接间隔搜索，因为存在中间 隔两个节点的情况

法2：直接采用递归的方法问题在于会超时，原因是访问相同的节点会产生重复计算，可以把子节点计算出来的MaxData存在node->val中(负数)

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
#define max(a,b)    (a>b)?a:b	

int rob(struct TreeNode* root){
    if(!root)   return 0;

    int l=rob(root->left);
    int r=rob(root->right);
    
    int sum1=0;          
    sum1=max(sum1,l+r);              //sum1不带根

    int l1=0,l2=0,r1=0,r2=0;
    if(root->left){
        if(root->left->left)    l1=rob(root->left->left);
        if(root->left->right)   l2=rob(root->left->right);
    }
    
    if(root->right){
        if(root->right->left)   r1=rob(root->right->left);
        if(root->right->right)  r2=rob(root->right->right);
    }
    
    int sum2=root->val;
    sum2=max(sum2,root->val+l1+l2+r1+r2);  //sum2带根节点
    
    return max(sum1,sum2);    
}

---

1. 动态规划即把多阶段过程转化为一系列单阶段问题，利用各阶段之间的关系，逐个求解（类似高中数学的数列） ↩︎