python数据结构之二叉树的建立实例
程序员文章站
2023-11-03 21:39:10
先建立二叉树节点，有一个data数据域，left，right 两个指针域复制代码 代码如下:# -*- coding: utf - 8 - *- &n...
先建立二叉树节点，有一个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:
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 也可以得到同样的效果