首页 > 编程 > Python > 正文

python数据结构之二叉树的建立实例

2020-02-23 05:22:26
字体:
来源:转载
供稿:网友

先建立二叉树节点,有一个data数据域,left,right 两个指针域
代码如下:
# -*- coding: utf - 8 - *-

                 
class TreeNode(object):

    def __init__(self, left=0, right=0, data=0):
        self.left = left
        self.right = right
        self.data = data

代码如下:
class BTree(object):

    def __init__(self, root=0):
        self.root = root

手动建立二叉树

代码如下:
node1 = TreeNode(data=1)
node2 = TreeNode(node1, 0, 2)
node3 = TreeNode(data=3)
node4 = TreeNode(data=4)
node5 = TreeNode(node3, node4, 5)
node6 = TreeNode(node2, node5, 6)
node7 = TreeNode(node6, 0, 7)
node8 = TreeNode(data=8)
root = TreeNode(node7, node8, 'root')

bt = BTree(root)

然后会生成下面的二叉树
代码如下:
# 生成的二叉树

# ------------------------
#          root
#       7        8
#     6
#   2   5
# 1    3 4
#
# -------------------------

除了 手动一个个的制定 node 节点,还可以创建一个 create 方法,接受用户输入添加二叉树节点。。。使用前续方式添加 ,代码如下:
代码如下:
# -*- coding: utf - 8 - *-

           
class TreeNode(object):

    def __init__(self, left=0, right=0, data=0):
        self.left = left
        self.right = right
        self.data = data

           
class BTree(object):

    def __init__(self, root=0):
        self.root = root

    def is_empty(self):
        if self.root is 0:
            return True
        else:
            return False

    def create(self):
        temp = input('enter a value:')
        if temp is '#':
            return 0
        treenode = TreeNode(data=temp)
        if self.root is 0:

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表