给定一个二叉树, 找到该树中两个指定节点的最近公共祖先

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先

三种情况:
1)、如果左边为空,右边不为空,则右边的第一个节点就为公共祖先
2)、如果右边为空,左边不为空,则左边的第一个节点就为公共祖先
3)、如果左右都有,则root就为公共祖先

根据前序遍历的方式来写

第一步:定义p、q,判断是二叉树否为空
第二步:如果p或q为root,直接返回root;如果不是,分别在左子树,右子树查找p、q
第三步:查找分三种情况,若左右子树都不为空,且如果p或q为root,直接返回root
若左子树不为空,右子树为空,则共同祖先就在左子树上,返回左子树
若右子树不为空,左子树为空,则共同祖先就在右子树上,返回右子树

class Solution5 {
    public TreeNode2 lowestCommonAncestor(TreeNode2 root, TreeNode2 p, TreeNode2 q) {
        if(root==null){
            return null;
        }

        if(root==p || root==q){
          return root;
        }

        TreeNode2 leftTree=lowestCommonAncestor(root.left,p,q);
        TreeNode2 rightTree=lowestCommonAncestor(root.right,p,q);

        if(leftTree!=null && rightTree!=null){
            return root;
        }

        if(leftTree !=null){
            return leftTree;
        }

        if(rightTree !=null){
            return rightTree;
        }
        return null;
    }



你可能感兴趣的:(java)