代码随想录刷题第22天|LeetCode235二叉搜索树的最近公共祖先、LeetCode701二叉搜索树中的插入操作 、LeetCode450删除二叉搜索树中的节点

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

题目链接：235、二叉搜索树的最近公共祖先

递归法：

注意返回值，如果没有返回值，可能会错过最近公共祖先。

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

迭代法：

注意最后的 return NULL；

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;
    }
};

2、LeetCode701二叉搜索树中的插入操作

题目链接：701、二叉搜索树中的插入操作

递归法（有返回值）：

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (root == NULL)
        {
            TreeNode* node = new TreeNode(val);
            return node;
        }

        if (root->val > val) root->left = insertIntoBST(root->left, val);
        if (root->val < val) root->right = insertIntoBST(root->right, val);

        return root;
    }
};

递归法（无返回值）：

class Solution {
public:
    TreeNode* parent;
    void traversal(TreeNode* root, int val)
    {
        if (root == NULL)
        {
            TreeNode* node = new TreeNode(val);
            if (parent->val > val) parent->left = node;
            else
            {
                parent->right = node;
            }
            return;
        }

        parent = root;
        if (root->val > val) traversal(root->left, val);
        if (root->val < val) traversal(root->right, val);
        return;
    }
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (root == NULL)
        {
            root = new TreeNode(val);  
        }
        traversal(root, val);
        return root;

    }
};

3、LeetCode450删除二叉搜索树中的节点

题目链接：450、删除二叉搜索树中的节点

分为5种情况：

• 第一种情况：没找到删除的节点，遍历到空节点直接返回了
• 找到删除的节点
  • 第二种情况：左右孩子都为空，直接删除节点， 返回NULL为根节点
  • 第三种情况：删除节点的左孩子为空，右孩子不为空，删除节点，右孩子补位，返回右孩子为根节点
  • 第四种情况：删除节点的右孩子为空，左孩子不为空，删除节点，左孩子补位，返回左孩子为根节点
  • 第五种情况：左右孩子节点都不为空，则将删除节点的左子树头结点（左孩子）放到删除节点的右子树的最左面节点的左孩子上，返回删除节点右孩子为新的根节点。

    class Solution {
    public:
        TreeNode* deleteNode(TreeNode* root, int key) {
            if (root == NULL) return root;
            if (root->val == key)
            {
                if (root->left == NULL && root->right == NULL)
                {
                    delete root;
                    return NULL;
                }
                else if (root->left == NULL)
                {
                    TreeNode* right = root->right;
                    delete root;
                    return right;
                }
                else if (root->right == NULL)
                {
                    TreeNode* left = root->left;
                    delete root;
                    return left;
                }
                else
                {
                    TreeNode* cur = root->right;
                    while(cur->left != NULL)
                    {
                        cur = cur->left;
                    }
                    cur->left = root->left;
                    TreeNode* tmp = root;
                    root = root->right;
                    delete tmp;
                    return root;
                    // TreeNode* node = root->right;
                    // delete root;
                    // return node;
                }
            }
    
            if (root->val > key) root->left = deleteNode(root->left, key);
            if (root->val < key) root->right = deleteNode(root->right, key);
            return root;
        }
    };