Leetcode337. 打家劫舍 III(C语言)
数据结构-树:算法与数据结构参考
题目:
在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。例:
输入: [3,4,5,1,3,null,1]
输出: 9
思路:
法1:动态规划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);
}
动态规划即把多阶段过程转化为一系列单阶段问题,利用各阶段之间的关系,逐个求解(类似高中数学的数列) ↩︎