树的定义
树状图是一种数据结构,它是由n(n>=1)个有限结点组成一个具有层次关系的集合。
实现树的一种方法可以是在每一个节点除数据外还有一些指针,使得该节点的每一个儿子都有一个指针指向它。然后,由于每个节点的儿子数可以变化并且事先不知道,因此在数据结构中建立到各儿子节点直接的链接是不可行的,因为这样会产生太多的浪费空间。
实现
树节点
class Node:
def __init__(self, Data):
self.data = Data
self.firstchild = None
self.nextsibling = None
# 设置树节点的值
def setData(self, newData):
self.data = newData
# 查看树节点的值
def getData(self):
return self.data
# 查看树节点的第一个儿子
def getFirstChild(self):
return self.firstchild
# 设置树节点的第一个儿子
def setFirstChild(self, newTree):
self.firstchild = newTree
# 查看树节点的下一个兄弟
def getNextSibling(self):
return self.nextsibling
# 设置树节点的下一个兄弟
def setNextSibling(self, newTree):
self.nextsibling = newTree
# 前序遍历
def pre_order_traversal(self):
print(self.getData(), end=' ')
if self.getFirstChild():
self.getFirstChild().pre_order_traversal()
if self.getNextSibling():
self.getNextSibling().pre_order_traversal()
elif self.getNextSibling():
self.getNextSibling().pre_order_traversal()
# 后序遍历
def post_order_traversal(self):
if self.getFirstChild():
self.getFirstChild().post_order_traversal()
print(self.getData(), end=' ')
else:
print(self.getData(), end=' ')
if self.getNextSibling():
self.getNextSibling().post_order_traversal()
树
class Tree:
def __init__(self, root = None):
self.root = root
def isEmpty(self):
return self.root is None
测试
# 创建树
t = Node(1)
t.setFirstChild(Node(2))
t.getFirstChild().setNextSibling(Node(3))
t.getFirstChild().setFirstChild(Node(4))
t.getFirstChild().getFirstChild().setNextSibling(Node(5))
t.getFirstChild().getNextSibling().setFirstChild(Node(6))
t.getFirstChild().getNextSibling().getFirstChild().setNextSibling(Node(7))
T = Tree(t)
print("树是否为空:",T.isEmpty()) # False
print("--------------------------")
print("前序遍历:")
T.root.pre_order_traversal() # 1 2 4 5 3 6 7
print()
print("--------------------------")
print("后序遍历")
T.root.post_order_traversal() # 4 5 2 6 7 3 1
print()
print("--------------------------")