第二十天 第六章 二叉树part07 235. 二叉搜索树的最近公共祖先 701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点

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

两种方法： 1.对所有二叉树都可以使用。

                   2.正对于搜索二叉树(利用其性质)。

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == q || root == p || root == NULL) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(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;     
    }
};
*/

class Solution {
public:
    TreeNode * traversal(TreeNode* root, TreeNode* p, TreeNode* q){
        if(root==NULL) return root;

        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) {       
          return traversal(root,  p,  q);
    }
};

701.二叉搜索树中的插入操作

用上层节点去接下层节点，即接新添加的节点。随后节点一步一步往根节点走的，最终返回根节点就行。需要多看讲解视频去理解。 

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

    }
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        return traversal(root,val);
    }
};

450.删除二叉搜索树中的节点

处理节点的左右子节点都不为空的细节很多。我们要将该节点(删除的节点)的左子树连接到右子树的最左子树。

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

        return traversal(root,key);
    }
};