LeetCode 236.二叉树的最近公共祖先(C++) 存储父节点

存储父节点思路

我们可以用哈希表存储所有节点的父节点，然后我们就可以利用节点的父节点信息从 p 结点开始不断往上跳，并记录已经访问过的节点，再从 q 节点开始不断往上跳，如果碰到已经访问过的节点，那么这个节点就是我们要找的最近公共祖先。
算法
从根节点开始遍历整棵二叉树，用哈希表记录每个节点的父节点指针。
从 p 节点开始不断往它的祖先移动，并用数据结构记录已经访问过的祖先节点。
同样，我们再从 q 节点开始不断往它的祖先移动，如果有祖先已经被访问过，即意味着这是 p 和 q 的深度最深的公共祖先，即 LCA 节点。

class Solution {
public:
    unordered_map<int, TreeNode*> fa;//用哈希表记录每个节点的父节点指针。
    unordered_map<int, bool> vis;
    void dfs(TreeNode* root){
        if (root->left != nullptr) {
            fa[root->left->val] = root;//根的左子节点的值为key root为其父节点指针
            dfs(root->left);
        }
        if (root->right != nullptr) {
            fa[root->right->val] = root;//根的左子节点的值为key root为其父节点指针
            dfs(root->right);
        }
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        fa[root->val] = nullptr;//根节点的值的父节点指针为空指针
        dfs(root);//用哈希表记录每个节点的父节点指针。
        while (p != nullptr) {
            vis[p->val] = true;//记录节点p的值key 的访问记录（true or false）
            p = fa[p->val];//把p->val的父节点赋值给指针p
        }
        while (q != nullptr) {
            if (vis[q->val]) return q;//若vis[q->val]已经访问过（即是true） 则返回q指针
            q = fa[q->val];//否则 继续寻找q的父节点
        }
        return nullptr;//其他情况返回空指针
    }
};

复杂度分析
时间复杂度：O(N)，其中 N 是二叉树的节点数。二叉树的所有节点有且只会被访问一次，从 p 和 q 节点往上跳经过的祖先节点个数不会超过 N，因此总的时间复杂度为 O(N)。
空间复杂度：O(N) ，其中 N 是二叉树的节点数。递归调用的栈深度取决于二叉树的高度，二叉树最坏情况下为一条链，此时高度为 N，因此空间复杂度为 O(N)，哈希表存储每个节点的父节点也需要 O(N) 的空间复杂度，因此最后总的空间复杂度为 O(N)。