236. 二叉树的最近公共祖先

最新推荐文章于 2024-07-30 16:12:13 发布

小草dym

最新推荐文章于 2024-07-30 16:12:13 发布

阅读量110

点赞数

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/qq_39368007/article/details/117340721

版权

数据结构与算法专栏收录该内容

40 篇文章 1 订阅

订阅专栏

 * 去以root为根节点的二叉树中查找p、q的最近公共祖先
 * ① 如果p、q同时存在于这棵二叉树中，就能成功返回它们的最近公共祖先
 * ② 如果p、q都不存在于这棵二叉树中，返回null
 * ③ 如果只有p存在于这棵二叉树中，返回p
 * ④ 如果只有q存在于这棵二叉树中，返回q

下面的这种写法是错误的

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null || root == p || root == q) return root;
        // 去以root.left为根节点的二叉树中查找p、q的最近公共祖先
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        // 去以root.right为根节点的二叉树中查找p、q的最近公共祖先
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if (left != null && right == null) {
            return left;
        }
        if (left == null && right != null) {
            return right;
        }
        return root;        
    }
}

在这里插入图片描述

正确的解法如下：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null || root == p || root == q) return root;
        // 去以root.left为根节点的二叉树中查找p、q的最近公共祖先
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        // 去以root.right为根节点的二叉树中查找p、q的最近公共祖先
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if (left != null && right != null) return root;
        return (left != null) ? left : right;      
    }
}