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

来源:互联网 发布:会计专用软件 编辑:程序博客网 时间:2024/05/06 12:11


解题思路:

递归求解,若为空树返回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));    }};


0 0