1. 程式人生 > >二叉樹的建立(python)

二叉樹的建立(python)

 說明

1、構造器,定義一個構造器,用來構造樹。

2、向樹中新增資料。

如圖我們定義了這樣一顆二叉樹,我們用程式碼來建立它

二叉樹

用程式碼實現,程式碼如下:

#coding=utf-8

class TreeNode(object):
    def __init__(self,data=None,left=None,right=None):
        self.data = data
        self.left = left
        self.right = right

    # 這一步是在每次呼叫某個結點時,自動呼叫.data的方法
    def __str__(self):
        return str(self.data)

# 方法一
A, B, C, D, E, F, G, H, I = [TreeNode(x) for x in 'ABCDEFGHI']
A.left, A.right = B, C
B.right = D
C.left, C.right = E, F
E.left = G
F.left,F.right = H, I
print C.right

# 方法二
A = TreeNode('A','B','C')
B = TreeNode('B',None,'D')
C = TreeNode('C','E','F')
E = TreeNode('E','G',None)
F = TreeNode('F','H','I')
print C.right