mirror of
https://github.com/krahets/hello-algo.git
synced 2025-12-19 07:17:54 +08:00
Fix the return type of binary search tree and avl tree
This commit is contained in:
@@ -107,10 +107,9 @@ class AVLTree
|
||||
}
|
||||
|
||||
/* 插入节点 */
|
||||
public TreeNode? insert(int val)
|
||||
public void insert(int val)
|
||||
{
|
||||
root = insertHelper(root, val);
|
||||
return root;
|
||||
}
|
||||
|
||||
/* 递归插入节点(辅助方法) */
|
||||
@@ -132,10 +131,9 @@ class AVLTree
|
||||
}
|
||||
|
||||
/* 删除节点 */
|
||||
public TreeNode? remove(int val)
|
||||
public void remove(int val)
|
||||
{
|
||||
root = removeHelper(root, val);
|
||||
return root;
|
||||
}
|
||||
|
||||
/* 递归删除节点(辅助方法) */
|
||||
@@ -162,7 +160,11 @@ class AVLTree
|
||||
else
|
||||
{
|
||||
// 子节点数量 = 2 ,则将中序遍历的下个节点删除,并用该节点替换当前节点
|
||||
TreeNode? temp = getInOrderNext(node.right);
|
||||
TreeNode? temp = node.right;
|
||||
while (temp.left != null)
|
||||
{
|
||||
temp = temp.left;
|
||||
}
|
||||
node.right = removeHelper(node.right, temp.val);
|
||||
node.val = temp.val;
|
||||
}
|
||||
@@ -174,18 +176,6 @@ class AVLTree
|
||||
return node;
|
||||
}
|
||||
|
||||
/* 获取中序遍历中的下一个节点(仅适用于 root 有左子节点的情况) */
|
||||
private TreeNode? getInOrderNext(TreeNode? node)
|
||||
{
|
||||
if (node == null) return node;
|
||||
// 循环访问左子节点,直到叶节点时为最小节点,跳出
|
||||
while (node.left != null)
|
||||
{
|
||||
node = node.left;
|
||||
}
|
||||
return node;
|
||||
}
|
||||
|
||||
/* 查找节点 */
|
||||
public TreeNode? search(int val)
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user