2023每日刷題(五十二)
Leetcode—337.打家劫舍III
算法思想
實現代碼
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:pair<int, int> dfs(TreeNode* root) {if(root == nullptr) {return {0, 0};}auto [left_rob, left_not_rob] = dfs(root->left);auto [right_rob, right_not_rob] = dfs(root->right);// 選該結點int rob = left_not_rob + right_not_rob + root->val;// 不選該結點int not_rob = max(left_rob, left_not_rob) + max(right_rob, right_not_rob);return {rob, not_rob};}int rob(TreeNode* root) {auto [rob, not_rob] = dfs(root);return max(rob, not_rob);}
};
運行結果
之后我會持續更新,如果喜歡我的文章,請記得一鍵三連哦,點贊關注收藏,你的每一個贊每一份關注每一次收藏都將是我前進路上的無限動力 !!!↖(▔▽▔)↗感謝支持!