二叉树
最近想弄清楚各种树结构的定义,用处,那么先从最简单的二叉树说起
关于二叉树的定义以及一些术语,这里不再赘述,直接代码实现,为后续搜索树以及多叉树提供基础
class BTreeNode:
def __init__(self, data=None, l_child=None, r_child=None, parent=None):
self.data = data
self.l_child = l_child
self.r_child = r_child
self.parent = parent
深度
求树的深度,可以利用遍历结果求出,但光求深度其实也很简单,主要思想是递归,取左右子树深度的最大值加1即为最后求得深度,所以比较适合用递归
这里实现一个特殊方法__len__,获取树的深度
def __len__(self):
"""求树的高度,默认空树高度为0,只有根节点时高度为1"""
# 空树
if self.data is None:
return 0
if self.l_child is None and self.r_child is None:
return 1
if self.l_child is None:
return len(self.r_child) + 1
if self.r_child is None:
return len(self.l_child) + 1
return max(len(self.l_child), len(self.r_child)) + 1
遍历
广度优先和深度优先搜索
层次遍历
利用广度优先思想,用一个队列维护每一层节点,逐层向下访问
def travel_level(self):
"""层次遍历"""
queue = [self]
while queue:
cur = queue.pop(0)
if cur is not None:
print(cur