DAY22——二叉树part8

vvvvvampire

已于 2023-06-15 00:39:42 修改

阅读量95

点赞数

文章标签：算法

于 2023-06-15 00:36:21 首次发布

本文链接：https://blog.csdn.net/vvvvvampire/article/details/131218539

版权

该文介绍了二叉搜索树的基本操作，包括如何找到两个节点的最近公共祖先，以及如何进行插入和删除节点的操作。在查找最近公共祖先时，利用了二叉搜索树的特性；插入操作采用递归方式，保持树的结构；删除节点时需考虑三种情况，包括节点无子节点、有一个子节点和有两个子节点的情况。

摘要由CSDN通过智能技术生成

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

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root.val > p.val && root.val > q.val) return lowestCommonAncestor(root.left, p, q);  //返回left
        if (root.val < p.val && root.val < q.val) return lowestCommonAncestor(root.right, p, q);  //返回right
        return root;
    }
}

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

不考虑改变树的结构，使用递归法

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root == null){
            return new TreeNode(val);
        }
        if(root.val < val){
            root.right = insertIntoBST(root.right, val);
        }
        if(root.val > val){
            root.left = insertIntoBST(root.left, val);
        }
        return root;
    }
}

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

class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
    if (root == null) return root;
    if (root.val == key) {
      if (root.left == null) {  //只有一侧的子节点，此时直接返回
        return root.right;
      } else if (root.right == null) {
        return root.left;
      } else {   //找到需要删除的节点
        TreeNode cur = root.right;
        while (cur.left != null) {  //找到比删除的节点大的数（最小大于）
          cur = cur.left;
        }
        cur.left = root.left;  //将删除节点的左子树赋给新找到的值
        root = root.right;  //直接返回右子树
        return root;
      }
    }
    if (root.val > key) root.left = deleteNode(root.left, key);
    if (root.val < key) root.right = deleteNode(root.right, key);
    return root;
  }
}