二叉树两个节点最近公共祖先的解法

最新推荐文章于 2022-06-24 13:35:29 发布

原创

最新推荐文章于 2022-06-24 13:35:29 发布 · 921 阅读

1 ·

CC 4.0 BY-SA版权

本文探讨如何解决二叉树中找到给定节点p和q的最近公共祖先的问题。通过定义一个递归函数，处理包括p、q在树中不同情况的边界条件，从而优雅地找到解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

假设有一个二叉树、根节点为TreeNode root、p节点和 q节点均在二叉树中、求p和q的最近公共祖先节点。

优雅的递归解法:

对于这个问题, 我们定义一个函数、但是这个函数扩充这个问题的定义

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q)

这个函数的扩充定义是什么呢？

虽然题目是保证p、q均在树中、但是我们为了实现递归解法、必须要考虑p、q有一个不在或者是均不在树中的情况来应对子树的情况。

扩充定义:

1、如果根节点为null、返回null

2、如果p在树中、q不在、该函数返回p; 如果p不在树中、q在、则返回q;

3、如果p、q均不在树中、返回null、

4、如果p、q均在树中、返回最近公共祖先。

至于为什么定义2是这样、完全是一种扩充自定义规则、也是一种巧妙地设计。

public class NearestCommonAncestor {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null){
            return null;
        }
        // 这一步的意思是:只要root节点和其中一个节点相等、那么即使另一个节点不在树中、也返回这个节点。
        if(root == p || root == q) {
            return root;
        }
        TreeNode lAns = lowestCommonAncestor(root.left, p, q);
        if(lAns == null){
            return lowestCommonAncestor(root.right, p, q);
        }