代码随想录算法训练营第31天|LeetCode235.二叉搜索树的最小公共祖先

代码随想录算法训练营第31天|LeetCode235.二叉搜索树的最小公共祖先

1、LeetCode235.二叉搜索树的最小公共祖先

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

代码随想录 (programmercarl.com)

第一想法：还是用普通二叉树最小公共祖先的做法。不知道怎么利用二叉搜索树的性质。

思路

因为是有序树，所以 如果 中间节点是 q 和 p 的公共祖先，那么中节点的数值 一定是在 [p, q]区间的 。即中节点 > p && 中节点 < q 或者 中节点 > q && 中节点 < p 。

因此，只要从上到下去遍历，遇到 cur节点是数值在[p, q]区间中则一定可以说明该节点cur就是p 和 q的公共祖先。

那一定是最小公共祖先吗？
如图所示，遍历到5。 如果 从节点5继续向左遍历，那么将错过成为p的祖先 ， 如果从节点5继续向右遍历则错过成为q的祖先 。
所以当我们从上向下去递归遍历，第一次遇到 cur节点是数值在[q, p]区间中，那么cur就是 q和p的最近公共祖先。

递归法

1. 参数和返回值
2. 停止条件
   cur为空，返回空
3. 单层逻辑
   root大，往左边遍历
   root小，往右边遍历
   cur在[p,q]或[q,p]区间内，返回cur。

注意:

如果递归函数有返回值，如何区分要搜索一条边，还是搜索整个树。

搜索一条边的写法：

if (递归函数(root->left)) return ;
if (递归函数(root->right)) return ;

搜索整个树写法：

left = 递归函数(root->left);
right = 递归函数(root->right);
left与right的逻辑处理;

代码
犯错：
单层遍历的逻辑弄混了，一开始写的是“root小，往左；root大，往右”。找了好一会儿这个bug。

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(root == NULL) 
            return NULL;
        if(root->val > p->val && root->val > q->val){
            //root大，往左边遍历
            return lowestCommonAncestor(root->left, p, q);
        }
        else if(root->val < p->val && root->val < q->val){
            //root小，往右边遍历
            return lowestCommonAncestor(root->right, p, q);
        }
        else{
            // root在区间内
            return 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 root;
    }
};

总结

这题主要在于思路。知道当root在区间内，那么root就是最近公共祖先的话，写代码就很简单了。