algorithm/hello-algo
1
0
Fork 0
mirror of https://github.com/krahets/hello-algo.git synced 2025-07-07 06:44:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

add binary_tree and avl_tree python code

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

32
codes/python/chapter_tree/binary_tree.py
View File

@ -1,10 +1,38 @@
'''
"""
File: binary_tree.py
Created Time: 2022-11-25
Author: Krahets (krahets@163.com)
'''
"""
import sys, os.path as osp
sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__))))
from include import *
""" Driver Code """
if __name__ == "__main__":
# 初始化二叉树
# 初始化节点
n1 = TreeNode(val=1)
n2 = TreeNode(val=2)
n3 = TreeNode(val=3)
n4 = TreeNode(val=4)
n5 = TreeNode(val=5)
n1.left = n2
n1.right = n3
n2.left = n4
n2.right = n5
print_tree(n1)
# 插入与删除结点
P = TreeNode(0)
# 在 n1 -> n2 中间插入节点 P
n1.left = P
P.left = n2
print_tree(n1)
# 删除结点
n1.left = n2
print_tree(n1)