二叉树的最近公共祖先(Java)

题目描述

给定一棵二叉树，找出树中两个给定节点的最低公共祖先(LCA)。
根据Wikipedia对LCA的定义:“在两个节点p和q之间定义的最低公共祖先是T中同时具有p和q作为后代的最低节点(在这里，我们允许一个节点作为其自身的后代)。”

解决

与找二叉搜索树的最近公共祖先类似，如果一个节点的左子树上有与p或q相等的节点且右子树上有与p或q相等的节点，说明此时该节点即为最近公共节点
代码：

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //递归出口
        if(root == null || root == p || root ==q)
            return root;
         //去该节点的左子树上找
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        //去该节点的右子树上找
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if(left == null){
        //左子树上没有，说明在右子树上
            return right;
        }else if(right == null){
        //右子树上没有，说明在左子树上
            return left;
        }
        //左右均有，说明该节点即为最近公共祖先
        return root;
    }