Day19：剑指 Offer68 二叉搜索树的最近公共祖先

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

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

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

题目关键

这是一颗二叉搜索树，而且所有的值都是唯一的，所以p、q可以直接根据val来区分他们在root的左子树还是右子树。所以就很简单了，总共就四种情况，1.p、q在root的异侧，所以root就是他们的最近公共祖先。2.q、p有其中一个是root，根据题目信息，自己可以当自己祖先，所以此时的root就是最近公共祖先。3.p、q在root的左子树，那么在遍历root->left。4.q、p在root的右子树，那么遍历root->right。时间复杂度为O(N)

方法一：递归

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        // 退出条件。当pq在root两侧时root就是最近共节点。
        // if ((root->val > p->val && root->val < q->val) || 
        // (root->val < p->val && root->val > q->val)) return root;
        // if (root->val == p->val || root->val == q->val) return root; 
        // 代码简化，以上两种情况都是返回root，所以在最后return就好
        // 当q、p都在左侧时，遍历至root的左边
        if (q->val < root->val && p->val < root->val) return lowestCommonAncestor(root->left, p, q);
        // 当pq在右侧时，遍历至root的右边。
        if (q->val > root->val && p->val > root->val)return lowestCommonAncestor(root->right, p, q);
        return root;
    }
};

方法二：迭代

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(p->val > q->val)
            swap(p, q);
        while(root != nullptr) {
            if(root->val < p->val) // p,q 都在 root 的右子树中
                root = root->right; // 遍历至右子节点
            else if(root->val > q->val) // p,q 都在 root 的左子树中
                root = root->left; // 遍历至左子节点
            else break;
        }
        return root;
    }
};

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

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

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

题目关键

现在是一颗普通的二叉树。还是延续上面的想法，p，q在异侧还是同侧。需要一个dfs遍历，碰到p,q就返回。
还是三部曲。
首先是退出条件。1.当越过了叶子节点，返回nullptr。2.root和qp其中一个相等时，返回root。
其次递推工作，1.开启遍历左节点，并将返回值记为left，遍历右节点返回值记为right。
最后是返回值，1.left和right同时为空，说明root左右子树都不包含pq，返回 nullptr。
2.同时不为空，pq在root两侧，root即是最近公共祖先。3.left为空，right不为空，返回right。这里面主要有两种情况，一种是当返回到总体时候，left为空，right不为空，说明pq都在右子树中，因为递归，所以此时root指向的就是最近公共祖先。另一种是局部的情况，遍历到一颗子树的左右次子树了，q在这个右边次子树，p则可能在另一颗子树上。
注意点：left和right是要根据每次递归有不同的值的，所以不应该将它定义成全局变量，这是错的。

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 && right == nullptr) return nullptr; // 1.
        if(left == nullptr) return right; // 3.
        if(right == nullptr) return left; // 4.
        return root; // 2. if(left != null and right != null)
    }
};