码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2019-10-05 18:34:44      阅读:109      评论:0      收藏:0      [点我收藏+]

标签:__init__   utf-8   put   用户输入   div   交互   btree   实例   python   

先建立二叉树节点,有一个data数据域,left,right 两个指针域

# coding:utf-8
class TreeNode(object):
    def __init__(self,left=None,right=None,data=0):
        self.left = left
        self.right = right
        self.data =data

建立树:

class BTree(object):
    def __init__(self,root=None):
        self.root = root

手动建立二叉树

if __name__ == "__main__":
    node1 = TreeNode(data=1)
    node2 = TreeNode(node1, None, 2)
    node3 = TreeNode(data=3)
    node4 = TreeNode(data=4)
    node5 = TreeNode(node3, node4, 5)
    node6 = TreeNode(node2, node5, 6)
    node7 = TreeNode(node6, None, 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=None,right=None,data=0):
        self.left = left
        self.right = right
        self.data =data

class BTree(object):
    def __init__(self,root=None):
        self.root = root

    def is_empty(self):
        return self.root == None

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

        treenode.left = self.create()
        treenode.right = self.create()

if __name__ == "__main__":
    bt = BTree()
    bt.create()

使用create创建二叉树

#运行文件 在交互解释器下面运行

bt = BTree()

bt.create()
enter a value:9
enter a value:7
enter a value:6
enter a value:2
enter a value:1
enter a value:#
enter a value:#
enter a value:#
enter a value:5
enter a value:3
enter a value:#
enter a value:#
enter a value:4
enter a value:#
enter a value:#
enter a value:#
enter a value:8
enter a value:#
enter a value:#

通过 create 也可以得到同样的效果

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

标签:__init__   utf-8   put   用户输入   div   交互   btree   实例   python   

原文地址:https://www.cnblogs.com/ceo-python/p/11625198.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!