leetcode 450
思路
对于要删除节点的操作,我们首先需要知道删除的节点有哪几种情况
- 没有要删除的节点
- 删除的节点为叶子节点
- 删除的节点左节点为空 右节点不为空
- 删除的节点做节点不为空 右节点为空
- 删除的节点左右节点都不为空
我们如果查找到当前节点的值是目标值,那么我们需要判断这些情况,然后进行处理,返回的值会在后面进行接收,比如示例中假设要删除的节点是7,7此时是叶子节点,那么我们就需要返回null, 原本6->left = 7 现在更新后变成 6->left = null
假设要删除的节点是6,那么我们发现6有右子树,没有左子树,那么我们就返回6的右子树(7),让6的父节点5指向7
如果要删除的节点是3,我们发现3的左右节点都有值
,那么我们需要把左右节点都重构,因为是二叉搜索树,所以根据特性,我们可以让3的左子树作为子树根节点也可以让3的右子树作为子树根节点,此时我们让3的右子树作为根节点,那么我们知道3的左子树的任何节点都是比3的右子树的任何节点都更小的,所以我们需要找到右子树的最小的节点,也就是一直向左边遍历到底,最后一个元素是右子树最小值,此时再让最小值的左子树为根节点3的左子树就重构完成
下图是删除操作的过程:可以看出我们先找到两个节点:删除元素7的左子树节点5,删除元素7的右子树内的最小值8,让8->left = 5,返回节点9(7->right),设置2->right = 9
图片来自代码随想录
实现
var deleteNode = function (root, val) {// 没找到删除的元素if (!root) return root;if (root.val === val) {if (!root.left && !root.right) {// 叶子节点return null} else if (root.left && !root.right) {// 左节点存在 右节点不存在return root.left} else if (!root.left && root.right) {// 左节点不存在,右节点存在return root.right;} else if (root.left && root.right) {// 左右节点都存在,重构子树return getCurNode(root);}}if(val < root.val){root.left = deleteNode(root.left,val)}if(val > root.val){root.right = deleteNode(root.right, val)}return root;
};
// 对于左右节点都存在的情况进行重新构造
var getCurNode = function (root) {let curNode = root.right;while (curNode.left) curNode = curNode.left;curNode.left = root.left;return root.right;
}