Leetcode 337. 打家劫舍 III

1、问题分析

题目链接：https://leetcode-cn.com/problems/house-robber-iii/
  本质上就是一个二叉树的遍历问题。代码我已经进行了详细的注释，理解应该没有问题，读者可以作为参考，如果看不懂(可以多看几遍)，欢迎留言哦！我看到会解答一下。

2、问题解决

  笔者以C++方式解决。

#include "iostream"

using namespace std;

#include "algorithm"
#include "vector"
#include "queue"
#include "set"
#include "map"
#include "string"
#include "stack"

/**
* Definition for a binary tree node.
* struct TreeNode {
*     int val;
*     TreeNode *left;
*     TreeNode *right;
*     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;

    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
private:
    // 保存访问过的节点能够盗取的最高金额
    // key 为：节点指针地址 和 父节点是否访问 的结合体
    map<pair<TreeNode *,int>, int> dp_map;
public:
    int rob(TreeNode *root) {
        // 如果树为空，则直接返回 0
        if (root == NULL) {
            return 0;
        }

        //  不抢劫根节点时再去抢劫左右子树
        int dontVisit = dealChen(root->left, 0) + dealChen(root->right, 0);
        // 抢劫根节点，再去抢劫左右子树
        int visited = dealChen(root->left, 1) + dealChen(root->right, 1) + root->val;
        // 选择其中较大的
        return max(dontVisit, visited);

    }

    /**
     * 根据父节点的状态抢劫树
     * @param root
     * @param flag 0 代表父节点没有被抢劫; 1 代表父节点被抢劫
     * @return
     */
    int dealChen(TreeNode *root, int flag) {
        // 如果节点为空，则直接返回 0
        if (root == NULL) {
            return 0;
        }

        // 首先查找该节点是否计算过，
        auto iterator1 = dp_map.find(make_pair(root,flag));
        // 如果计算过，则直接返回计算过的值
        if (iterator1 != dp_map.end()) {
            return iterator1->second;
        }

        // 不抢劫本节点，直接抢劫左右子树
        int left = dealChen(root->left, 0);
        int right = dealChen(root->right, 0);
        // 不抢劫本节点获取的最高金额
        int dontVisit = left + right;

        // 如果父节点已经被抢劫过，则直接返回 不抢劫本节点获取的最高金额
        if (flag == 1) {
            // 同时保存该值
            dp_map.insert(make_pair(make_pair(root, flag), dontVisit));
            return dontVisit;
        }

        // 抢劫该节点之后，再去抢劫左右子树
        int leftVisited = dealChen(root->left, 1);
        int rightVisited = dealChen(root->right, 1);
        // 抢劫本节点 和 不抢劫本节点 选择其中较大的值
        int visited = max(dontVisit, root->val + leftVisited + rightVisited);
        // 同时保存该值
        dp_map.insert(make_pair(make_pair(root, flag), visited));
        // 返回计算结果
        return visited;
    }
};

int main() {

//    TreeNode *pNode1 = new TreeNode(3);
//    TreeNode *pNode2 = new TreeNode(2);
//    TreeNode *pNode3 = new TreeNode(3);
//    TreeNode *pNode4 = new TreeNode(3);
//    TreeNode *pNode5 = new TreeNode(1);
//
//    pNode1->left = pNode2;
//    pNode1->right = pNode3;
//    pNode2->right = pNode4;
//    pNode3->right = pNode5;

    TreeNode *pNode11 = new TreeNode(3);
    TreeNode *pNode12 = new TreeNode(4);
    TreeNode *pNode13 = new TreeNode(5);
    TreeNode *pNode14 = new TreeNode(1);
    TreeNode *pNode15 = new TreeNode(3);
    TreeNode *pNode16 = new TreeNode(1);

    pNode11->left = pNode12;
    pNode11->right = pNode13;
    pNode12->left = pNode14;
    pNode12->right = pNode15;
    pNode13->right = pNode16;



    Solution *pSolution = new Solution;
    int i = pSolution->rob(pNode11);
    cout << i << endl;
    system("pause");
    return 0;
}

运行结果

有点菜，有时间再优化一下。

3、总结

  难得有时间刷一波LeetCode, 这次做一个系统的记录，等以后复习的时候可以有章可循，同时也期待各位读者给出的建议。算法真的是一个照妖镜，原来感觉自己也还行吧，但是算法分分钟教你做人。前人栽树，后人乘凉。在学习算法的过程中，看了前辈的成果，受益匪浅。
感谢各位前辈的辛勤付出，让我们少走了很多的弯路！
哪怕只有一个人从我的博客受益，我也知足了。
点个赞再走呗！欢迎留言哦！