打劫房屋 III-LintCode

在上次打劫完一条街道之后和一圈房屋之后，窃贼又发现了一个新的可以打劫的地方，但这次所有的房子组成的区域比较奇怪，聪明的窃贼考察地形之后，发现这次的地形是一颗二叉树。与前两次偷窃相似的是每个房子都存放着特定金额的钱。你面临的唯一约束条件是：相邻的房子装着相互联系的防盗系统，且当相邻的两个房子同一天被打劫时，该系统会自动报警。

算一算，如果今晚去打劫，你最多可以得到多少钱，当然在不触动报警装置的情况下。

样例：

3
/ \
2   3
\   \
3   1

窃贼最多能偷窃的金钱数是 3 + 3 + 1 = 7.

3
/ \
4   5
/ \   \
1   3   1

窃贼最多能偷窃的金钱数是 4 + 5 = 9.

思路：

修改二叉树每个节点的值，返回根节点值为最后结果。

对于如下矩阵

a
/ \
b   c
/ \  /\
d  e f g

a->val=max(b+c,d+e+f+g)

用此方法遍历每个节点，得到最终结果。

#ifndef C535_H
#define C535_H
#include<iostream>
using namespace std;
class TreeNode{
public:
int val;
TreeNode *left, *right;
TreeNode(int val)
{
this->val = val;
this->left = this->right = NULL;
}
};
class Solution {
public:
/*
* @param root: The root of binary tree.
* @return: The maximum amount of money you can rob tonight
*/
int houseRobber3(TreeNode * root) {
// write your code here
if (root == NULL)
return 0;
int a = houseRobber3(root->left);
int b = houseRobber3(root->right);
int num1 = 0;
int num2 = 0;
int num3 = 0;
int num4 = 0;
if (root->left != NULL)
{
if (root->left->left != NULL)
num1 = root->left->left->val;
if (root->left->right != NULL)
num2 = root->left->right->val;
}
if (root->right != NULL)
{
if (root->right->left != NULL)
num3 = root->right->left->val;
if (root->right->right != NULL)
num4 = root->right->right->val;
}
root->val = maxVal(a + b, num1 + num2 + num3 + num4 + root->val);
return root->val;
}
int maxVal(int a, int b)
{
return a > b ? a : b;
}
};
#endif