牛客BM38 在二叉树中找到两个节点的最近公共祖先

问题：

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

描述：

        给定一棵二叉树(保证非空)以及这棵树上的两个节点对应的val值 o1 和 o2，请找到 o1 和 o2 的最近公共祖先节点。

        数据范围：树上节点数满足1≤n≤10^5  , 节点值val满足区间 [0,n)

        要求：时间复杂度 O(n)O(n)

        注：本题保证二叉树中每个节点的val值均不相同。

        如当输入{3,5,1,6,2,0,8,#,#,7,4},5,1时，二叉树{3,5,1,6,2,0,8,#,#,7,4}如下图所示：

        所以节点值为5和节点值为1的节点的最近公共祖先节点的节点值为3，所以对应的输出为3。节点本身可以视为自己的祖先

解题方法：

1、非递归

思路：

        要想找到两个节点的最近公共祖先节点，我们可以从两个节点往上找，每个节点都往上走，一直走到根节点，那么根节点到这两个节点的连线肯定有相交的地方，如果是从上往下走，那么最后一次相交的节点就是他们的最近公共祖先节点。我们就以找6和7的最近公共节点来画个图看一下：

        我们看到6和7公共祖先有5和3，但最近的是5。我们只要往上找，找到他们第一个相同的公共祖先节点即可，但怎么找到每个节点的父节点呢，我们只需要把每个节点都遍历一遍，然后顺便记录他们的父节点存储在Map中。我们先找到其中的一条路径，比如6→5→3，然后在另一个节点往上找，由于7不在那条路径上，我们找7的父节点是2，2也不在那条路径上，我们接着往上找，2的父节点是5，5在那条路径上，所以5就是他们的最近公共子节点。

        其实这里我们可以优化一下，我们没必要遍历所有的结点，我们一层一层的遍历（也就是BFS），只需要这两个节点都遍历到就可以了，比如上面2和8的公共结点，我们只需要遍历到第3层，把2和8都遍历到就行了，没必要再遍历第4层了。（BFS就是一层一层的遍历，如下图所示）

代码：

    public int lowestCommonAncestor(TreeNode root, int o1, int o2) {
        //记录遍历到的每个节点的父节点。
        Map<Integer, Integer> parent = new HashMap<>();
        //辅助队列，用于层次遍历
        Queue<TreeNode> queue = new LinkedList<>();
        //根节点没有父节点，给他默认一个值（键为当前节点子节点的值，值为父节点的值）
        parent.put(root.val, Integer.MIN_VALUE);
        //先将根节点入队列
        queue.add(root);
        //直到两个节点都找到为止。
        while (!parent.containsKey(o1) || !parent.containsKey(o2)) {
            //队列是一边进一边出，这里poll方法是出队，
            TreeNode node = queue.poll();
            //记录每一个节点的父子关系，并进行遍历
            if (node.left != null) {
                //左子节点不为空，记录下他的父节点
                parent.put(node.left.val, node.val);
                //左子节点不为空，把它加入到队列中
                queue.add(node.left);
            }
            //右节点同上
            if (node.right != null) {
                parent.put(node.right.val, node.val);
                queue.add(node.right);
            }
        }
        Set<Integer> ancestors = new HashSet<>();
        //记录下o1和他的祖先节点，从o1节点开始一直到根节点。
        while (parent.containsKey(o1)) {
            ancestors.add(o1);
            o1 = parent.get(o1);
        }
        //查看o1和他的祖先节点是否包含o2节点，如果不包含再看是否包含o2的父节点
        while (!ancestors.contains(o2))
            o2 = parent.get(o2);
        return o2;
    }

 2、递归

思路：

        目标值为树上的两个节点值，所以目标值在树上，那么目标值可能位于树的根节点两侧，也可能位于根节点的一侧，如果位于根节点两侧，那么最近公共祖先节点位于最后一个能同时在它的左右子树中找到目标值的根节点，而如果位于一侧的话，那么最近公共祖先节点位于最先能找到目标值的节点，并且此时在数的另外一棵子树中找不到目标值。我们以下图来演示一下：

        我们以找6和7的最近公共节点看一下：当来到根节点为5的子树时，我们能在它的左子树找到一个目标节点，在它的右子树找到另外一个目标节点，那么此时我们就可以得知节点5就是它的6，7的最近公共祖先节点。

        再以找5和6的最近公共节点看一下：当来到根节点为3的左子树时，我们找到一个目标值5，然后再去根节点的右子树找看能否找到另外一个值，如果不能找到，那么我们可以肯定的是这个目标值就位于以节点5为根节点的这棵树中，此时的最近公共祖先节点就是节点5。而每次在子树中寻找目标值都是一个子问题，所以可以使用递归，在当前树的子树中寻找目标值。

步骤：
1）以根节点为起始节点，判断当前节点是否为目标值

2）以前序遍历的方式进行递归，先进入其左子树，再进入其右子树

3）若在当前的左子树中寻找到目标节点，则返回，并记录当前节点的值，若不存在，则记录为-1

4）再遍历其右子树，寻找另一个目标值，若该目标值存在，则该根节点为最近公共祖先节点，若当前根节点的右子树不存在会依次回调其父节点的右子树，直到来到整棵树的根节点，在谁的左右子树中分别找到两个目标值，则该节点为最近公共祖先节点。若是未找到另一个目标值，则该目标值位于第一个目标值的子树中，所以第一个目标值为最近公共祖先节点。
 

代码：

public class Solution {
    public int lowestCommonAncestor (TreeNode root, int p, int q) {
        //树为空
        if(root == null) 
            return -1;
        //判断节点是否为目标值
        if(root.val == p || root.val == q){
            return root.val;
        }
        //左子树寻找目标值
        int left = lowestCommonAncestor(root.left, p, q);
        //右子树寻找目标值
        int right = lowestCommonAncestor(root.right, p, q);
        //如果当前根节点的左右子树都存在目标值，则当前节点为最近公共祖先节点
        if(left != -1 && right != -1) 
            return root.val;
        //若目标值都位于左子树，则先找到的目标值为最近公共祖先节点
        if(left != -1) 
            return left;
        //若目标值都位于右子树，则先找到的目标值为最近公共祖先节点
        if(right != -1) 
            return right;
        return -1;  
    }
}