lintcode-最近公共祖先-88

给定一棵二叉树,找到两个节点的最近公共父节点(LCA)。

最近公共祖先是两个节点的公共的祖先节点且具有最大深度。


样例

对于下面这棵二叉树

 4 / \ 3 7 / \ 5 6 

LCA(3, 5) = 4

LCA(5, 6) = 7

LCA(6, 7) = 7

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
   
    TreeNode *lowestCommonAncestor(TreeNode *root, TreeNode *A, TreeNode *B) {
        if(!root)                 
            return NULL; 
        if(root==A)      
            return A;
        if(root==B)
            return B;
       
        TreeNode *L=lowestCommonAncestor(root->left,A,B);
        TreeNode *R=lowestCommonAncestor(root->right,A,B);
    
        if(L&&R)  //如果A和B都在root的子节点中,返回root
            return root;

/*      
        if(!L&&!R)        //如果A和B都不在root的子节点中,返回NULL
            return NULL;  
        if(L&&!R)       //如果A在B不在,返回A,告诉上一层A节点已经在该节点的子节点中
             return L;           
        if(!L&&R)
             return R;
                      //实际上这几行代码可以省略 可以包含在最后一段代码中
*/
        return L?L:R;      
    }
};


你可能感兴趣的:(lintcode-最近公共祖先-88)