337. House Robber III (C++实现)

337. House Robber III (C++实现)_第1张图片

解题思路:

递归求解,若为空树返回0;

若只有一个根结点就返回根结点的值;

否则,若根结点有孩子,两种抢法:抢根结点!不抢根结点,抢根结点的孩子子树!取最大值(选抢的多的方案)!

(注:抢根结点,就不能抢根结点的孩子!但可以抢根结点的孩子的孩子~~~)

C++代码:


/**
 * 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) {
        if(root==NULL)return 0;
        else if(root->left==NULL&&root->right==NULL)return root->val;
        else if(root->left==NULL)return max((root->val)+rob(root->right->left)+rob(root->right->right),rob(root->right));
        else if(root->right==NULL)return max((root->val)+rob(root->left->left)+rob(root->left->right),rob(root->left));
        else return max((root->val)+rob(root->right->left)+rob(root->right->right)+rob(root->left->left)+rob(root->left->right),rob(root->right)+rob(root->left));
    }
};


你可能感兴趣的:(leetcode,C/C++)