加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

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

发布时间:2020-12-16 20:15:55 所属栏目:Python 来源:网络整理
导读:先建立二叉树节点,有一个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 复制代码 代码如

先建立二叉树节点,有一个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,2)
node3 = TreeNode(data=3)
node4 = TreeNode(data=4)
node5 = TreeNode(node3,node4,5)
node6 = TreeNode(node2,node5,6)
node7 = TreeNode(node6,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,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:
            self.root = treenode

        treenode.left = self.create()
        treenode.right = self.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 也可以得到同样的效果

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读