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