代码随想录Day20——二叉树

二叉搜索树求最近公共祖先

题目理解

二叉搜索树是有序的，如果是左中右排的话，只要在有序数组中查找两数之间的根节点就可以了。不需要遍历整棵树，找到就可以。

代码块

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

二叉搜索树中的插入操作

题目理解

在已知的一个二叉搜索树中插入一个节点，使插入操作之后该树依然是二叉搜索树。

思路

代码

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* deleteNode(TreeNode* root, int key) {
      if(root==NULL)  return root;
      if(root->val==key){
        if(root->left==NULL&&root->right==NULL) return NULL;
        if(root->left!=NULL&&root->right==NULL) return root->left;
        if(root->left==NULL&&root->right!=NULL) return root->right;
        else{
        TreeNode* curr=root->right;
        while(curr->left!=NULL)  {curr=curr->left;}
        curr->left = root->left;
        TreeNode*tmp=root;
        root =root->right;
        tmp->left=curr->left;
        delete tmp;
        return root;
        }
      }
      if(root->val>key) root->left=deleteNode(root->left,key);
      if(root->val<key) root->right=deleteNode(root->right,key);
      return root;
    }
};