LeetCode—236.二叉树的最近公共祖先(Lowest Common Ancestor of a Binary Tree)——分析及代码（C++）

一、题目

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

百度百科中最近公共祖先的定义为：“对于有根树 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 为不同节点且均存在于给定的二叉树中。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 递归

（1）思路

可将函数设计为：若树中存在节点 p 或 q，返回先等于 p 或 q 的节点；若不存在，返回空指针。
则从根节点开始，对左右子树依次递归：
1）若左、右子树均返回非空节点，该节点为最近公共祖先；
2）若左子树返回空节点，说明最近公共祖先在右子树；
3）若右子树返回空节点，说明最近公共祖先在左子树。

（2）代码

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == nullptr || root == p || root == q)
            return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (left == nullptr)
            return right;
        else if (right == nullptr)
            return left;
        else
            return root;
    }
};

（3）结果

执行用时 ：20 ms, 在所有 C++ 提交中击败了 90.18% 的用户；
内存消耗 ：16.4 MB, 在所有 C++ 提交中击败了 95.12% 的用户。

2. 中序遍历

（1）思路

通过栈实现中序遍历，并结合栈大小判断节点的深度关系。
记录第一个遇到的 p 或 q 节点，在此基础上寻找第二个节点，若通过栈发现到达了所记录节点的父节点，则更新节点。

（2）代码

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL)
            return root;
        stack<TreeNode*> sta;
        TreeNode* node = root;
        TreeNode* ans = NULL;//记录当前最近祖先
        int stalen = 0;//记录当前最近祖先对应的栈大小
        while (node != NULL || !sta.empty()) {//中序遍历
            while (node != NULL) {
                sta.push(node);
                node = node->left;
            }
            node = sta.top();
            if (ans != NULL && sta.size() < stalen) {//若通过栈发现到达了所记录节点的父节点，则更新节点
                ans = node;
                stalen = sta.size();
            }
            if (node == p || node == q)
                if (ans == NULL) {//记录第一个遇到的 p 或 q 节点
                    ans = node;
                    stalen = sta.size();
                }
                else//遇到第二个节点，此时记录的祖先节点即最近公共祖先
                    return ans;
            sta.pop();
            node = node->right;
        }
        return ans;
    }
};

（3）结果

执行用时 ：20 ms, 在所有 C++ 提交中击败了 90.18% 的用户；
内存消耗 ：16.4 MB, 在所有 C++ 提交中击败了 97.70% 的用户。

三、其他

暂无。