题目描述
给定一个二叉搜索树的根节点root和一个值key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
思路
如题目描述中所说,先找到节点,再删除节点。
由于要保持二叉搜索树的特点,即左节点的值均小于本节点的值,右节点的值均大于本节点的值。
因此,在删除对应节点后,要注意将该节点的左子树接到右子树的最左侧。
此外,如果删除的是根节点,则需要将根节点的子节点作为新的根节点返回。
代码
首先,在main函数中进行讨论,如果要删除的是根节点root,则直接调用insert函数,返回新的根节点;如果要删除的不是根节点,则通过深搜dfs函数进行遍历,若遇到子节点值为key,则调用insert,返回新的子节点。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
if( root==null ) return root;
if( root.val==key ){
return insert( root.left , root.right );
}
dfs( root, key );
return root;
}
public TreeNode dfs( TreeNode root, int key ){
if( root == null ) return null;
if( root.left!=null && root.left.val== key ){
root.left = insert( root.left.left , root.left.right );
return null;
}
else if( root.right!=null && root.right.val== key ){
root.right = insert( root.right.left , root.right.right );
return null;
}
dfs( root.left , key );
dfs( root.right , key );
return root;
}
public TreeNode insert( TreeNode left , TreeNode right ){
if( left==null && right==null ) return null;
else if( left==null ) return right;
else if( right==null ) return left;
if( right.left!=null ){
insert( left , right.left );
}
else{
right.left = left;
}
return right;
}
}
改进
看了leetcode官解后,发现自己在dfs函数中没有利用到二叉搜索树的性质,即子节点的val不为key时,应通过val与key的大小关系,决定递归左子树还是右子树。