[ 二叉树 ] 二叉树的最近公共祖先

236. 二叉树的最近公共祖先 - 力扣（LeetCode） (leetcode-cn.com)

二叉树的最近公共祖先

递归

• 自底向上
• 回溯
• 后序遍历

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        // 自底向上 == 回溯 == 后序遍历
        if (!root || root == p || root == q) return root;
        
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        
        // 如果left都和right是p或q的祖先，那么root就是最小公共祖先
        if (left && right) return root;
        
        // 如果left为空，那么p或q在right内， right是p或q的一个祖先
        else if (!left && right) return right;
        
        // 如果right为空，那么p或q在left内， left是p或q的一个祖先
        else if (left && !right) return left;
        
        // 否则找不到pq祖先
        else return nullptr;
    }
};

---

235. 二叉搜索树的最近公共祖先 - 力扣（LeetCode） (leetcode-cn.com)

二叉搜索树的最近公共祖先

递归

• BST性质使得祖先值和子孙值有关系利用这层关系可以快速查找

• 自顶向下，我们遇到的第一个值在pq之间的结点就是最近公共祖先。

• 先序遍历

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        //p, q 都在左子树，到左子树处理
        if (root->val > p->val && root->val > q->val) 
            return lowestCommonAncestor(root->left, p, q);
        //p, q 都在右子树，到右子树处理
        else if (root->val < p->val && root->val < q->val) 
            return lowestCommonAncestor(root->right, p, q);
        //p, q 分别在左右子树中，那么返回此时结点
        else return root;
    }
};

非递归

• 自顶向下，我们遇到的第一个值在pq之间的结点就是最近公共祖先。

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        while (root) {
            //p, q 都在左子树，到左子树处理
            if (root->val > p->val && root->val > q->val) root = root->left;
            //p, q 都在右子树，到右子树处理
            else if (root->val < p->val && root->val < q->val) root = root->right;
            //p, q 分别在左右子树中，那么返回此时结点
            else return root;
        }
        return nullptr;
    }
};