add binary_tree and avl_tree python code

This commit is contained in:
a16su
2022-12-20 15:52:00 +08:00
parent 2a1bb23990
commit 9eac1275f6
9 changed files with 967 additions and 22 deletions
+104
View File
@@ -82,6 +82,23 @@ comments: true
=== "Python"
```python title="binary_search_tree.py"
def search(self, num):
"""
查找结点
"""
cur = self.get_root()
# 循环查找,越过叶结点后跳出
while cur is not None:
# 目标结点在 root 的右子树中
if cur.val < num:
cur = cur.right
# 目标结点在 root 的左子树中
elif cur.val > num:
cur = cur.left
# 找到目标结点,跳出循环
else:
break
return cur
```
@@ -228,7 +245,37 @@ comments: true
=== "Python"
```python title="binary_search_tree.py"
def insert(self, num):
"""
插入结点
"""
root = self.get_root()
# 若树为空,直接提前返回
if root is None:
return None
cur = root
pre = None
# 循环查找,越过叶结点后跳出
while cur is not None:
# 找到重复结点,直接返回
if cur.val == num:
return None
pre = cur
if cur.val < num: # 插入位置在 root 的右子树中
cur = cur.right
else: # 插入位置在 root 的左子树中
cur = cur.left
# 插入结点 val
node = TreeNode(num)
if pre.val < num:
pre.right = node
else:
pre.left = node
return node
```
=== "Go"
@@ -483,7 +530,64 @@ comments: true
=== "Python"
```python title="binary_search_tree.py"
def remove(self, num):
"""
删除结点
"""
root = self.get_root()
# 若树为空,直接提前返回
if root is None:
return None
cur = root
pre = None
# 循环查找,越过叶结点后跳出
while cur is not None:
# 找到待删除结点,跳出循环
if cur.val == num:
break
pre = cur
if cur.val < num: # 待删除结点在 root 的右子树中
cur = cur.right
else: # 待删除结点在 root 的左子树中
cur = cur.left
# 若无待删除结点,则直接返回
if cur is None:
return None
# 子结点数量 = 0 or 1
if cur.left is None or cur.right is None:
# 当子结点数量 = 0 / 1 时, child = null / 该子结点
child = cur.left or cur.right
# 删除结点 cur
if pre.left == cur:
pre.left = child
else:
pre.right = child
# 子结点数量 = 2
else:
# 获取中序遍历中 cur 的下一个结点
nex = self.min(cur.right)
tmp = nex.val
# 递归删除结点 nex
self.remove(nex.val)
# 将 nex 的值复制给 cur
cur.val = tmp
return cur
def min(self, root):
"""
获取最小结点
"""
if root is None:
return root
# 循环访问左子结点,直到叶结点时为最小结点,跳出
while root.left is not None:
root = root.left
return root
```
=== "Go"