以下程序作为编写二叉树程序,方便自己调试使用
1 # coding=utf-8 2 class TreeNode(object): 3 def __init__(self, data=None, left=None, right=None): 4 self.data = data 5 self.left = left 6 self.right = right 7 8 # 这一步是在每次调用某个结点时,自动调用.data的方法 9 #当使用print输出对象的时候,只要自己定义了__str__(self)方法, 10 # 那么就会打印从在这个方法中return的数据 11 def __str__(self):# 返回一个字符串,是对该对象的描写 12 return str(self.data) 13 14 # 方法一 15 A, B, C, D, E, F, G, H, I = [TreeNode(x) for x in 'ABCDEFGHI'] 16 A.left, A.right = B, C 17 B.right = D 18 C.left, C.right = E, F 19 E.left = G 20 F.left, F.right = H, I 21 print(C.right) 22 23 # 方法二 24 A = TreeNode('A', 'B', 'C') 25 B = TreeNode('B', None, 'D') 26 C = TreeNode('C', 'E', 'F') 27 E = TreeNode('E', 'G', None) 28 F = TreeNode('F', 'H', 'I') 29 print(C.right)
参考连接