【JZ-68-II】二叉树的最近公共祖先（深度优先遍历、递归）

最新推荐文章于 2022-10-03 08:36:00 发布

子木_98

最新推荐文章于 2022-10-03 08:36:00 发布

阅读量172

点赞数

分类专栏：编程练习

本文链接：https://blog.csdn.net/qq_38001377/article/details/115307895

版权

编程练习专栏收录该内容

74 篇文章 1 订阅

订阅专栏

这篇博客介绍了如何在非二叉搜索树中找到两个节点的最近公共祖先，通过深度优先遍历算法实现。在无法利用节点值进行比较的情况下，递归地检查左右子节点，并根据返回值判断最近公共祖先。算法的时间复杂度为O(n)，空间复杂度为O(n)，适用于任意二叉树。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

算法思路

与【JZ-68-I】的区别在于二叉树不再是二叉搜索树了，也就是说我们无法利用节点的值的大小关系来判断节点的位置关系了，可以采用 深度优先遍历。

递归终止条件：

越过了叶节点：直接返回 $n u l l$
$r o o t$ 等于p、q：返回 $r o o t$

递推过程：

开启左子节点的递归，返回值记为 $l e f t$
开启右子节点的递归，返回值记为 $r i g h t$

返回值（情况1.可以合并到3.和4.内）：

$l e f t$ 、 $r i g h t$ 均为空：返回 $n u l l$ ；
$l e f t$ 、 $r i g h t$ 均不为空：说明p、q分别在 $r o o t$ 的左右子树中， $r o o t$ 就是最近的公共祖先，返回 $r o o t$ ；
$l e f t$ 为空， $r i g h t$ 不为空：说明p、q都不在 $r o o t$ 的左子树中（可能是均在右子树中，也有可能是其中一个在右子树中），返回 $r i g h t$ （此时的 $r i g h t$ 可能指向最近公共祖先，也有可能指向在右子树中的那一个节点）；
$l e f t$ 不为空， $r i g h t$ 为空：与 3. 同理

具体代码

/**
 * 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;
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if(left == null)return right;
        if(right == null)return left;
        return root;
    }
}