Leetcode 337. 打家劫舍 III

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, 這次做一個系統的記錄,等以後複習的時候可以有章可循,同時也期待各位讀者給出的建議。算法真的是一個照妖鏡,原來感覺自己也還行吧,但是算法分分鐘教你做人。前人栽樹,後人乘涼。在學習算法的過程中,看了前輩的成果,受益匪淺。
感謝各位前輩的辛勤付出,讓我們少走了很多的彎路!
哪怕只有一個人從我的博客受益,我也知足了。
點個贊再走唄!歡迎留言哦!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章