2022-12-20 15:52:00 +08:00
|
|
|
"""
|
2022-11-25 02:04:38 +08:00
|
|
|
File: binary_tree.py
|
2022-12-27 18:34:12 +08:00
|
|
|
Created Time: 2022-12-20
|
|
|
|
|
Author: a16su (lpluls001@gmail.com)
|
2022-12-20 15:52:00 +08:00
|
|
|
"""
|
2022-11-25 02:04:38 +08:00
|
|
|
|
|
|
|
|
import sys, os.path as osp
|
2022-12-20 15:52:00 +08:00
|
|
|
|
2022-11-25 02:04:38 +08:00
|
|
|
sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__))))
|
|
|
|
|
from include import *
|
|
|
|
|
|
2022-12-20 15:52:00 +08:00
|
|
|
|
|
|
|
|
""" Driver Code """
|
|
|
|
|
if __name__ == "__main__":
|
2022-12-27 19:33:58 +08:00
|
|
|
""" 初始化二叉树 """
|
2022-12-20 15:52:00 +08:00
|
|
|
# 初始化节点
|
|
|
|
|
n1 = TreeNode(val=1)
|
|
|
|
|
n2 = TreeNode(val=2)
|
|
|
|
|
n3 = TreeNode(val=3)
|
|
|
|
|
n4 = TreeNode(val=4)
|
|
|
|
|
n5 = TreeNode(val=5)
|
2022-12-27 19:33:58 +08:00
|
|
|
# 构建引用指向（即指针）
|
2022-12-20 15:52:00 +08:00
|
|
|
n1.left = n2
|
|
|
|
|
n1.right = n3
|
|
|
|
|
n2.left = n4
|
|
|
|
|
n2.right = n5
|
2022-12-27 19:24:43 +08:00
|
|
|
print("\n初始化二叉树\n")
|
2022-12-20 15:52:00 +08:00
|
|
|
print_tree(n1)
|
|
|
|
|
|
2022-12-27 19:33:58 +08:00
|
|
|
""" 插入与删除结点 """
|
2022-12-20 15:52:00 +08:00
|
|
|
P = TreeNode(0)
|
|
|
|
|
# 在 n1 -> n2 中间插入节点 P
|
|
|
|
|
n1.left = P
|
|
|
|
|
P.left = n2
|
2022-12-27 19:24:43 +08:00
|
|
|
print("\n插入结点 P 后\n")
|
2022-12-20 15:52:00 +08:00
|
|
|
print_tree(n1)
|
|
|
|
|
# 删除结点
|
|
|
|
|
n1.left = n2
|
2023-01-08 19:03:22 +08:00
|
|
|
print("\n删除结点 P 后\n")
|
2022-12-20 15:52:00 +08:00
|
|
|
print_tree(n1)
|
