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

一、递归版本

思想:假设根结点为root,其中给定的两个结点分别为A和B,它们分别都不为null。如果当前结点p为null,那么直接返回null,如果当前结点p是给定的结点中的其中一个结点,那么直接返回当前结点p(如果p是根结点,程序一次就返回了,下面的递归也不会出现)。如果当前节点不是A和B中的一个,那么需要分别去查找p的左右子树,看看是否包含A或者B,查询左右子树后,如果查询左子树和查询右子树的结果都不为null,说明当前结点p就是最近的公共祖先。否则,如果查询左子树结果为null,那么返回查询右子树的结果。反之,返回查询左子树的结果。

public static TreeNode getParent(TreeNode root, TreeNode node1,TreeNode node2) {
        if(root == null || node1 == null ||  node2 == null) return null;
        //这里可以换成if(root == node1 || root == node2),我只是为了方便测试才这样写
        if(root.val == node1.val || root.val == node2.val) return root;
        TreeNode left = getParent(root.left,node1,node2);
        TreeNode right = getParent(root.right,node1,node2);
        //如果左右子树都能找到,那么当前节点就是最近的公共祖先节点
        if(left != null && right != null) return root;
        //如果左子树上没有,那么返回右子树的查找结果
        if(left == null) return right;
        //否则返回左子树的查找结果
        else return left;
    }

二、非递归版本

思想:这里提供一种思路,代码就不演示了。因为这个思想的实现代码和查找从根到某个结点的路径的思想一样。首先,我们找出一条从根到A的路径保存到栈或者数组,相同的方式找到一条从根到B的路径也保存到栈或者数组(参考我的另一篇博客:查找从根到某结点的路径)。然后开始遍历这两个数组,找到第一次下标相同但是值不同的那么数的下标,这个下标的前一个数就是最近公共祖先。我举个例子:

假设A为结点3,B为结点4,那么从根到A的路径为9---8---6---3,从根到B的路径为9---7---4,遍历这个两个数组,发现9 == 9,而8 != 7,所以9就是最近公共祖先。

原文地址:https://www.cnblogs.com/neuzk/p/9487301.html