文章目录
- 题目描述
- 思路
- 代码
题目描述
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
示例 1:
输入:root = [5,3,6,2,4,null,7], key = 3
输出:[5,4,6,2,null,null,7]
解释:给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。
一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
另一个正确答案是 [5,2,6,null,4,null,7]。
示例 2:
输入: root = [5,3,6,2,4,null,7], key = 0
输出: [5,3,6,2,4,null,7]
解释: 二叉树不包含值为 0 的节点
示例 3:
输入: root = [], key = 0
输出: []
提示:
节点数的范围 [0, 104].
-105 <= Node.val <= 105
节点值唯一
root 是合法的二叉搜索树
-105 <= key <= 105
进阶: 要求算法时间复杂度为 O(h),h 为树的高度。
思路
在代码中详细注释了
代码
class Solution {//使用递归的方法public TreeNode deleteNode(TreeNode root, int key) {//本方法采用的是:删除root的时候让左子树充当root,也就是示例一中的第二种答案//递归出口if (root==null){return null;}//如果找到了key,需要删除root这个节点,需要左右子树调整,所以选择采用后序遍历的方式,能够利用左右子树返回的结果TreeNode leftChild = deleteNode(root.left,key);//得到左子树的根节点TreeNode rightChild = deleteNode(root.right,key);//得到右子树的根节点//调整root的左右子树root.left = leftChild;root.right =rightChild;if (root.val==key){//找到key//如果这个节点是叶子节点就直接删除就行了,也就是直接返回nullif (leftChild==null&&rightChild==null){return null;}//左子树为空就返回右子树if (leftChild==null){return rightChild;}//右子树为空就返回左子树if (rightChild==null){return leftChild;}else {//如果右子树不为空就对子树的结构做出调整TreeNode cur = rightChild;//找到右子树中最左下方的节点,因为需要让这个节点的左孩子指向leftChild的右孩子//可以拿[5,3,6,2,4,null,7] key=5 这个案例在纸上画一下,删除5之后,4需要挂在6的左孩子的地方才符合二叉搜索树while (cur.left!=null){cur = cur.left;}cur.left = leftChild.right;}//删除根节点,左子树成为根节点leftChild.right = rightChild;return leftChild;}return root;}
}