剑指 Offer 68 - II. 二叉树的最近公共祖先(DFS递归、迭代非递归)2

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

百度百科中最近公共祖先的定义为：“对于有根树 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。因为根据定义最近公共祖先节点可以为节点本身。

解法一：后序遍历(DFS)，迭代

先递归到底，再回溯

复杂度分析：
时间复杂度 O(N)： 其中 N 为二叉树节点数；最差情况下，需要递归遍历树的所有节点。
空间复杂度 O(N)： 最差情况下，递归深度达到 N ，系统使用O(N) 大小的额外空间。

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 && right == null) return null; // 1.

        if(left == null) return right; // 3.

        if(right == null) return left; // 4.
        
        return root; // 2. if(left != null and right != null)

    }
}

解法二：迭代、非递归

使用哈希表记录每个节点的父节点，当找到p时利用哈希表找到他的父节点并标记，不断向上直到跟根节点，找到q时仍然利用哈希表找到它的父节点并查看是否已被标记，没有标记则继续向上，标记则返回该节点
算法：
1.从根节点开始遍历二叉树，用哈希表记录每个节点和其父节点
2.从p开始不断找到它的父节点并标为已访问，直到根节点
3.从q开始找到它的父节点（包括q）查看是否已被标记，被标记则返回
时间N： 遍历所有节点
空间N： map大小为n

class Solution {
    Map <TreeNode, TreeNode> map = new HashMap<>();
    Set <TreeNode> set = new HashSet<>();
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        dfs(root);
        while(p != null)
        {
            set.add(p);//p也有可能是结果
            p = map.get(p);
        }
        while(q != null)
        {
            if(set.contains(q))
                return q;
            q = map.get(q);
        }
        return null;
    }
    
    public void dfs(TreeNode root)
    {
        
        if(root.left != null)
        {
            map.put(root.left, root);
            dfs(root.left);
        }
        if(root.right != null)
        {
            map.put(root.right, root);
            dfs(root.right);
        }
        
    }
}