450.删除二叉搜索树中的节点
递归三部曲:
1、确定参数和返回值:
class Solution{
public TreeNode deleteNode(TreeNode root, int key){
}
}
2、确定终止条件
if(root == null){
return root;
}
3、确定单层遍历的逻辑:
//如果当前值等于要删除的值
if(root.val == key){
//如果当前节点没有左子节点,直接返回右子节点
if(root.left == null){
return root.right;
}
//如果当前节点没有右子节点,直接返回左子节点
else if(root.right == null){
return root.left;
}
//如果当前节点有2个子节点
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;
235. 二叉搜索树的最近公共祖先
class Solution {
// 定义一个方法,用于在二叉搜索树中寻找两个节点的最近公共祖先
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
// 使用while循环进行遍历
while (true) {
// 如果当前节点的值大于p和q的值,则最近公共祖先在左子树中
if (root.val > p.val && root.val > q.val) {
root = root.left; // 移动到左子节点
// 如果当前节点的值小于p和q的值,则最近公共祖先在右子树中
} else if (root.val < p.val && root.val < q.val) {
root = root.right; // 移动到右子节点
// 如果当前节点的值介于p和q的值之间,或者等于p或q的值,说明当前节点就是最近公共祖先
} else {
break; // 退出循环
}
}
// 返回最近公共祖先节点
return root;
}
}
class Solution {
// 定义一个方法,用于在二叉搜索树中插入一个新的节点
public TreeNode insertIntoBST(TreeNode root, int val) {
// 如果根节点为空,则创建一个新的节点作为根节点并返回
if (root == null) return new TreeNode(val);
// 保存根节点的引用,以便最后返回
TreeNode newRoot = root;
// 定义一个指针用于跟踪前一个节点
TreeNode pre = root;
// 使用while循环寻找插入位置
while (root != null) {
// 将当前节点赋值给pre
pre = root;
// 如果当前节点的值大于插入值,则移动到左子节点
if (root.val > val) {
root = root.left;
// 如果当前节点的值小于插入值,则移动到右子节点
} else if (root.val < val) {
root = root.right;
}
}
// 找到插入位置后,根据pre节点的值决定插入在左子节点还是右子节点
if (pre.val > val) {
pre.left = new TreeNode(val);
} else {
pre.right = new TreeNode(val);
}
// 返回更新后的根节点
return newRoot;
}
}