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


?做题思路or感想:

  • 递归一条边的写法:

    if (dfs(root->left))return;
    if (dfs(root->right))return;
    
  • 搜索整棵树的写法

    left = dfs(root->left);
    right = dfs(root->right);
    //left与right的逻辑处理
    

在递归函数又返回值的情况下:

  • 如果搜索一条边,那么在递归函数的返回值不为空的时候,立刻返回
  • 如果搜索整棵树,则直接用变量leftright保存返回值,接下来的逻辑就是用变量leftright来处理当前节点

方法一:就当作一颗普通二叉树来做

用后序遍历从底部开始往上遍历,因为是要找最近公共祖先,所以要搜索整棵树!

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 == p && right == q || left == q && right == p)return root;
        //不是,则继续传递递归的信息
        else if (!left && right)return right;
        else if (left && !right)return left;
        else return nullptr;
    }
};

但这种方法完全没有用到二叉搜索树的特性,暴珍天物啊

方法二:利用二叉搜索树的性质

如果cur节点的数值在[p, q]区间,则说明该节点cur就是最近公共祖先

明白上面的话后,就直接照做就好了

但是可以利用二叉搜索树的特性,如果节点值大于这个区间,就往左子树跑,如果小于这个区间,就往右子树跑

class Solution {
public:
    bool flag = 1;
    TreeNode* result = nullptr;
    void dfs(TreeNode* cur, TreeNode* max, TreeNode* min) {
        if (flag == 1) {	//这个flag可有可无,我自己习惯加的,用来控制递归的进行与否
            if (cur == nullptr)return;	//递归中止条件,其实这里也是可有可无,因为不可能遍历到空节点的
            if (cur->val <= max->val && cur->val >= min->val) {	//这里注意是大于等于和小于等于,题目里有说祖先节点可以是那两个节点的本身,故是闭区间[p, q]
                result = cur;
                flag = 0;
                return;
            }
            //根据二叉搜索树的特性
            if (cur->val > max->val)dfs(cur->left, max, min);
            if (cur->val < min->val)dfs(cur->right, max, min);
        }
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (q->val > p->val)swap(p, q);	//细节,这里必须让p是大的那个!当然也可以让q是大的那个,随自己喜欢
        dfs(root, p, q);
        return result;
    }
};