leetcode笔记：House Robber III

一. 题目描述

The thief has found himself a new place for his thievery again. There is only one entrance to this area, called the “root.” Besides the root, each house has one and only one parent house. After a tour, the smart thief realized that “all houses in this place forms a binary tree”. It will automatically contact the police if two directly-linked houses were broken into on the same night.

Determine the maximum amount of money the thief can rob tonight without alerting the police.

Example 1:

     3    / \   2   3    \   \      3   1

Maximum amount of money the thief can rob = 3 + 3 + 1 = 7.

Example 2:

     3    / \   4   5  / \   \  1   3   1

Maximum amount of money the thief can rob = 4 + 5 = 9.

二. 题目分析

该题是House Robber II和House Robber的后续。

题目的意思是，小偷找到了一个新的偷盗场所。这片区域只有一个入口，叫做“根”。除了根以外，每个屋子有且仅有一个父屋子。在踩点之后盗贼发现，所有的房间构造形成了一棵二叉树。如果两个直接相连的屋子在同时被盗窃，就会惊动警察。

编写函数判断盗贼在不惊动警察的情况下最多可以偷到的金钱数目。测试用例如题目描述。

针对该题，可使用深度优先搜索来解决。对于当前节点，只有盗窃和不盗窃两种操作，这取决于当前节点的子节点的状态，可用两个变量表示：pre和curr：

pre：当前节点（屋子）不偷盗时，所能获得的收益；取决于在该节点的两个子节点被偷盗时的收益之和。

curr：当前节点（屋子）偷盗时，所能获得的收益；由于相邻节点不能同时偷盗否则会引来警察，所以两个子节点不能偷盗，此时收益等于：父节点->val + 左子节点->pre + 右子节点->pre。

比较两种收益pre和curr，取较大者作为当前节点的最大收益。

三. 示例代码

/** * Definition for a binary tree node. * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public:    struct Money {        int pre;  // 该节点不偷，收益为子节点偷的情况下的收益和        int curr; // 该节点处偷，加上两子节点不偷时的收益        Money():pre(0), curr(0){}    };    int rob(TreeNode* root) {        Money sum = dfs(root);        return sum.curr;    }    Money dfs(TreeNode* root)    {        if (root == NULL) return Money();        Money leftMoney = dfs(root->left);   // 当前节点的左子节点收益情况        Money rightMoney = dfs(root->right); // 当前节点的右子节点收益情况        Money sumMoney;        sumMoney.pre = leftMoney.curr + rightMoney.curr; // 当前节点不偷        sumMoney.curr = max(sumMoney.pre, root->val + leftMoney.pre + rightMoney.pre);        return sumMoney;    }};

四. 小结

该题属于DFS的经典题目。