Leetcode C++《熱題 Hot 100-68》337. 打家劫舍 III

Leetcode C++《熱題 Hot 100-68》337. 打家劫舍 III

1. 題目

在上次打劫完一條街道之後和一圈房屋後，小偷又發現了一個新的可行竊的地區。這個地區只有一個入口，我們稱之爲“根”。 除了“根”之外，每棟房子有且只有一個“父“房子與之相連。一番偵察之後，聰明的小偷意識到“這個地方的所有房屋的排列類似於一棵二叉樹”。 如果兩個直接相連的房子在同一天晚上被打劫，房屋將自動報警。

計算在不觸動警報的情況下，小偷一晚能夠盜取的最高金額。

示例 1:

輸入: [3,2,3,null,3,null,1]

 3
/ \

2 3
\ \
3 1

輸出: 7
解釋: 小偷一晚能夠盜取的最高金額 = 3 + 3 + 1 = 7.

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/house-robber-iii
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

2. 思路

• 非常經典的題目，樹的遍歷以及動態規劃的結合
• 每個節點可以選擇或者不選擇，最後返回max

3. 代碼

/**
 * 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:
    int rob(TreeNode* root) {
        //動態規劃的題目
        pair<int, int> res = dfs(root);
        return max(res.first, res.second);
    }

    pair<int, int> dfs(TreeNode* root){  // first: choose; second:not choose
        if (root == NULL)
            return make_pair(0,0);
        pair<int, int> res;
        pair<int, int> left_res = dfs(root->left);
        pair<int, int> right_res = dfs(root->right);
        res.first = root->val + left_res.second + right_res.second;
        res.second = max(left_res.second, left_res.first) +max(right_res.second, right_res.first);
        return res;
    }
};