leetcode_House Robber III

class Solution {
public:
map<TreeNode *,int> zero;
map<TreeNode *,int> one;
int add(TreeNode * root ,int index)
{
if(root==NULL) return 0;
int value;
if(index==0)
{
if(zero[root]!=0) return zero[root];
else
{
value=max(add(root->left,0)+add(root->right,1),add(root->left,0)+add(root->right,0));
value=max(value,add(root->left,1)+add(root->right,1));
value=max(value,add(root->left,1)+add(root->right,0));
zero[root]=value;
return value;
}
}
else if(index==1)
{
if(one[root]!=0) return one[root];
else
{
value=add(root->left,0)+add(root->right,0)+root->val;
one[root]=value;
return value;
}
}
return 0;
}
int rob(TreeNode* root) {
return max(add(root,0),add(root,1));
}
};