二叉搜索树的最近公共祖先
leetcode链接:leetcode.cn/problems/lo…
利用二叉树的特性,如果p和q节点在当前节点的左右两边,说明当前节点一定是p和q节点的最近公共祖先
var lowestCommonAncestor = function(root, p, q) {
if (root === null) return null;
if (root.val > p.val && root.val > q.val) {
return root.left = lowestCommonAncestor(root.left, p, q);
}
if (root.val < p.val && root.val < q.val) {
return root.right = lowestCommonAncestor(root.right, p, q);
}
return root;
};
二叉搜索树中的插入操作
leetcode链接:leetcode.cn/problems/in…
根据搜索二叉树的特性,如果要在一棵二叉搜索树中插入一个节点,这个节点一定可以是叶子节点
var insertIntoBST = function(root, val) {
// 遇到叶子节点,可以插入新的节点了
if (!root) {
let newNode = new TreeNode(val);
return newNode;
}
if (val < root.val) {
root.left = insertIntoBST(root.left, val);
}
if (val > root.val) {
root.right = insertIntoBST(root.right, val);
}
return root;
};
删除二叉搜索树中的节点
leetcode链接:leetcode.cn/problems/de…
删除比插入复杂,需要考虑的情况比较多:
- 找不到要删除的节点
- 要删除的是叶子节点,节点的左边为空,右边也为空
- 要删除的节点左边不为空,右边为空
- 要删除的节点左边为空,右边不为空
- 要删除的节点左边不为空,右边不为空
var deleteNode = function(root, key) {
if (root === null) return null;
// 找到要删除的节点了
if (root.val === key) {
if (root.left === null && root.right === null) {
// 删除叶子节点操作
return null;
} else if (root.left && root.right === null) {
return root.left;
} else if (root.left === null && root.right) {
return root.right;
} else {
// 第5种情况,将右子树往上继承,左子树放在右子树的最左边
let cur = root.right;
while (cur.left) {
cur = cur.left;
}
cur.left = root.left;
return root.right;
}
} else if (root.val > key) {
// 往左边找要删除的节点
root.left = deleteNode(root.left, key);
} else {
// 往右边找要删除的节点
root.right = deleteNode(root.right, key);
}
return root;
};