*LeetCode 236 二叉树的最近公共祖先（简单）

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

例如，给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]

示例 1:

输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出: 3
解释: 节点 5 和节点 1 的最近公共祖先是节点 3。

示例 2:

输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出: 5
解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。

说明:

• 所有节点的值都是唯一的。
• p、q 为不同节点且均存在于给定的二叉树中。

思路与代码

自己想的方法

dfs遍历二叉树，用path记录走过的节点路径，再分别用两个list记录到两个节点的路径。最后比较两个路径的公共节点即可。

class Solution {
    List<TreeNode> list_p;
    List<TreeNode> list_q;
    List<TreeNode> path = new ArrayList();
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {

        dfs(root,p,q);

        TreeNode res = null;
        int size = Math.min(list_p.size(), list_q.size());
        for (int i = 0; i < size; i++) {
            if (list_p.get(i) == list_q.get(i)) res = list_q.get(i);
            else return res;
        }
        return res;
    }

    public void dfs(TreeNode node, TreeNode p, TreeNode q) {
        if (node == null) return;
        
        path.add(node);
        
        if (node == p) list_p = new ArrayList(path);   
        if (node == q) list_q = new ArrayList(path);
        
        dfs(node.left, p, q);
        dfs(node.right, p, q);
        path.remove(path.size()-1);
        
    }
}

推荐方法

当我们用递归去做这个题时不要被题目误导，应该要明确一点
这个函数的功能有三个：给定两个节点 p 和 q

• 如果 p 和 q 都存在，则返回它们的公共祖先；
• 如果只存在一个，则返回存在的一个；
• 如果 p 和 q 都不存在，则返回NULL

class Solution {

    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // 该函数返回值有以下情况
        // 如果p和q都在root树下 则返回的为最近公共组选
        // 如果只有p或q一个节点在root树下，返回的就是那个节点的祖先，即该节点（节点本身就是自己的祖先）
        // 如果两节点都不在root树下 说明root属下没有这两个节点的祖先

        if (root == null || root == q || root == p) return root;

        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);

        if (left == null && right == null) return null;
        if (left == null) return right;
        if (right == null) return left;
        return root; 
    }

}