Fix the return type of binary search tree and avl tree

This commit is contained in:
krahets
2023-04-14 05:47:20 +08:00
parent 9c9c8b7574
commit f7ae9c8a02
24 changed files with 247 additions and 451 deletions
+6 -15
View File
@@ -94,9 +94,8 @@ class AVLTree {
}
/* 插入节点 */
TreeNode? insert(int val) {
void insert(int val) {
root = insertHelper(root, val);
return root;
}
/* 递归插入节点(辅助方法) */
@@ -117,9 +116,8 @@ class AVLTree {
}
/* 删除节点 */
TreeNode? remove(int val) {
void remove(int val) {
root = removeHelper(root, val);
return root;
}
/* 递归删除节点(辅助方法) */
@@ -141,7 +139,10 @@ class AVLTree {
node = child;
} 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;
}
@@ -153,16 +154,6 @@ class AVLTree {
return node;
}
/* 获取中序遍历中的下一个节点(仅适用于 root 有左子节点的情况) */
TreeNode? getInOrderNext(TreeNode? node) {
if (node == null) return node;
// 循环访问左子节点,直到叶节点时为最小节点,跳出
while (node!.left != null) {
node = node.left;
}
return node;
}
/* 查找节点 */
TreeNode? search(int val) {
TreeNode? cur = root;