二叉树的公共祖先问题

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

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

示例 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 。因为根据定义最近公共祖先节点可以为节点本身。

示例 3：

输入：root = [1,2], p = 1, q = 2
输出：1

提示：

• 树中节点数目在范围 [2, 105] 内。
• -109 <= Node.val <= 109
• 所有 Node.val 互不相同 。
• p != q
• p 和 q 均存在于给定的二叉树中。

以上是普通二叉树的？如果是二叉搜索树又怎么处理呢？

class Solution {

public:

    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {

        if(root == NULL ) return NULL;  //这两个打破递归循环的逻辑，也就是函数的出口

        if(root == q || root== p) return root; //说明找到了一个p or q，返回当前的一个根节点，而不是null

       

        TreeNode* left = lowestCommonAncestor(root->left,p,q);

        TreeNode* right =lowestCommonAncestor(root->right,p,q);

        //中

        if(left!=NULL && right!= NULL) return root;// 进入递归的是root->left，返回的也是root->left，因此当满足条件是返回root

            //下面这个逻辑是递归逻辑

        if(left!=NULL && right ==NULL) return left;

        else if(right !=NULL && left == NULL) return right;

        else return NULL;

    }

};

我给它进行了注释！主要是围绕递归三部曲！

1.返回值和参数

2.打破递归的出口（条件）

3.单层的循环条件逻辑

而对于搜索二叉树的做法！是这样的！这样是错的，我陷入了一个错误的想法！

class Solution {

public:

    //先写一种情况，q->val < q->val   既 公共祖先的值在 [p,q]之间

    TreeNode* first_action(TreeNode* root, TreeNode* p, TreeNode* q) {

    if(root == NULL) return NULL ;

    if(root == q || root == p) return root;

    TreeNode* left = NULL;

    TreeNode* right = NULL;

    if(root->left != NULL && root->left->val >= p->val && root->left->val <= q->val) {

        left = first_action(root->left, p, q);

    } //进入左子树递归的条件

    if(root->right != NULL && root->right->val >= p->val && root->right->val <= q->val) {

        right = first_action(root->right, p, q);

    } //进入右子树递归的条件

    if(left != NULL && right != NULL) return root; //中

    // 单层

    if(left != NULL && right == NULL) return left;

    else if(left == NULL && right != NULL) return right;

    else return NULL;

}

TreeNode* sencen_action(TreeNode* root, TreeNode* p, TreeNode* q) {

    if(root == NULL) return NULL ;

    if(root == q || root == p) return root;

    TreeNode* left = NULL;

    TreeNode* right = NULL;

    if(root->left != NULL && root->left->val >= q->val && root->left->val <= p->val) {

        left = sencen_action(root->left, p, q);

    } //进入左子树递归的条件

    if(root->right != NULL && root->right->val >= q->val && root->right->val <= p->val) {

        right = sencen_action(root->right, p, q);

    } //进入右子树递归的条件

    if(left != NULL && right != NULL) return root; //中

    // 单层

    if(left != NULL && right == NULL) return left;

    else if(left == NULL && right != NULL) return right;

    else return NULL;

}

 

    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {

        TreeNode* node ;

        if(p->val < q->val) node=first_action(root,p,q);

        else node=sencen_action(root,p,q);

        return node;

    }

};

只要从上到下去遍历，遇到 cur节点是数值在[p, q]区间中则一定可以说明该节点cur就是p 和 q的公共祖先。 那问题来了，一定是最近公共祖先吗？所以当我们从上向下去递归遍历，第一次遇到 cur节点是数值在[q, p]区间中，那么cur就是 q和p的最近公共祖先。因为对应二叉搜索树来说，从上到下去遍历，只要第一次出现在这个区间的节点就一定是最近的公共祖先！因为他是有序的！

左边比根节点小，右边比根节点大！所以说只要出现了在[q,p区间里]，要是往左边走了，就会出现当前的节点大小值为范围[p,q),导致再也无法再遇见q了，同理往右边走也是同一个道理！！，往右边走的话，当前的节点大小值为范围（q,q+1)，导致公共祖先的节点【p,q】无法在相遇！因此，只要明白了这一点，我相信手到擒来！

而迭代法会更简单！

迭代代码如下：一直找符合这个范围的root！！！

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        while(root) {
            if (root->val > p->val && root->val > q->val) {
                root = root->left;
            } else if (root->val < p->val && root->val < q->val) {
                root = root->right;
            } else return root;
        }
        return NULL;
    }
};