記事ディレクトリ
トピック:337. House Robbery III
泥棒は前回、通りと家の輪を強奪した後、盗難が可能な新しい場所を発見しました。このエリアへの入り口は1つだけです。これを「ルート」と呼びます。「ルート」に加えて、各家には1つだけ「親」家が接続されています。いくつかの偵察の後、巧妙な泥棒は「この場所の家はすべて二分木のように配置されている」ことに気づきました。直接接続された2つの家が同じ夜に強奪されると、家は自動的に警察に電話します。
泥棒がアラームをトリガーせずに夜通し盗むことができる最大量を計算します。
例1:
输入: [3,2,3,null,3,null,1]
3
/ \
2 3
\ \
3 1
输出: 7
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
例2:
输入: [3,4,5,1,3,null,1]
3
/ \
4 5
/ \ \
1 3 1
输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.
出典:LeetCode
リンク:https ://leetcode-cn.com/problems/house-robber-iii
著作権はLeetCode が所有しています。商用転載については、正式な許可書にご連絡ください。非商用転載については、出典を明記してください。
基本的な考え方:DFS
- 現在のノードには、スチールするかスチールしないかの2つのオプションがあります。
- 盗む:その左または右の子ノードのどちらも盗まれることができません
- スチールなし:左および右の子ノードがスチールするか、スチールしないか
/**
* 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:
map<TreeNode *, vector<int>> m;
int rob(TreeNode* root) {
int res1, res2;
res1 = fun(root, false);
res2 = fun(root, true);
return res1 > res2 ? res1 : res2;
}
int fun(TreeNode* root, bool flag){
if(root == NULL)
return 0;
if(m.count(root) && m[root][flag] != INT_MAX)
return m[root][flag];
m[root] = vector<int>(2, INT_MAX);
int res1 = 0, res2 = 0;
int temp = fun(root->left, true) + fun(root->right, true);
if(flag == true){
//当前节点能偷
res1 = max(root->val + fun(root->left, false) + fun(root->right, false), temp);
}
else{
//当前节点不能偷
res2 = temp;
}
m[root][flag] = res1 > res2 ? res1 : res2;
return m[root][flag];
}
};