代码随想录算法训练营第二十四天 |二叉搜索树的最近公共祖先、二叉搜索树中的插入操作、删除二叉搜索树中的节点
补打卡,对二叉树不断的加深理解。
二叉搜索树的最近公共祖先link
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
//利用二叉搜索树的特性:根节点的左子树的值小于根节点的值,根节点的值小于右子树的值
//只要我们寻找的节点大于一个节点,同时小于一个节点,则为公共祖先
while (true) {
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 {
break;
}
}
return root;
}
}
二叉搜索树中的插入操作link
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if (root == null)
return new TreeNode(val);
//设置临时节点,用于对二叉树进行操作
TreeNode t = root;
while (true) {
if (t.val > val && t.left == null) {
t.left = new TreeNode(val);
break;
}
if (t.val < val && t.right == null) {
t.right = new TreeNode(val);
break;
}
if (t.val > val) {
t = t.left;
} else if (t.val < val) {
t = t.right;
}
}
return root;
}
}
删除二叉搜索树中的节点link
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
这个题有一定难度。
视频推荐:link
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;
else if (root.left != null && root.right == null)
return root.left;
else if (root.left == null && root.right != null)
return root.right;
else {
//都不为空的情况下,让右子树作为根部,则剩余的左子树放在原先右子树的左叶子节点的左边
TreeNode cur = root.right;
while (cur.left != null)
cur = cur.left;
//到现在,cur处于原先右子树的左叶子节点处
cur.left = root.left;
return root.right;
}
}
if (root.val > key)
root.left = deleteNode(root.left, key);
if (root.val < key)
root.right = deleteNode(root.right, key);
return root;
}
}
希望对您有帮助。