教你如何使用Python实现二叉树结构及三种遍历教你如何使用Python实现二叉树

一：代码实现

class TreeNode:"""节点类"""def __init__(self, mid, left=None, right=None):self.mid = midself.left = leftself.right = right# 树类class Tree:"""树类"""def __init__(self, root=None):self.root = rootdef add(self, item):# 将要添加的数据封装成一个node结点node = TreeNode(item)if not self.root:self.root = nodereturnqueue = [self.root]while queue:cur = queue.pop(0)if not cur.left:cur.left = nodereturnelse:queue.append(cur.left)if not cur.right:cur.right = nodereturnelse:queue.append(cur.right)tree = Tree()tree.add(0)tree.add(1)tree.add(2)tree.add(3)tree.add(4)tree.add(5)tree.add(6)

二：遍历在上述树类代码基础上加遍历函数，基于递归实现。

文章图片

先序遍历：

先序遍历结果是：0 -> 1 -> 3 -> 4 -> 2 -> 5 -> 6

# 先序遍历def preorder(self, root, result=[]):if not root:returnresult.append(root.mid)self.preorder(root.left, result)self.preorder(root.right, result)return resultprint("先序遍历")print(tree.preorder(tree.root))"""先序遍历[0, 1, 3, 4, 2, 5, 6]"""

中序遍历：

中序遍历结果是：3 -> 1 -> 4 -> 0 -> 5 -> 2 -> 6

# 中序遍历def inorder(self, root, result=[]):if not root:return resultself.inorder(root.left, result)result.append(root.mid)self.inorder(root.right, result)return resultprint("中序遍历")print(tree.inorder(tree.root))"""中序遍历3, 1, 4, 0, 5, 2, 6]"""

后续遍历

后序遍历结果是：3 -> 4 -> 1 -> 5 -> 6 -> 2 -> 0

# 后序遍历def postorder(self, root, result=[]):if not root:return resultself.postorder(root.left, result)self.postorder(root.right, result)result.append(root.mid)return resultprint("后序遍历")print(tree.postorder(tree.root))"""后序遍历[3, 4, 1, 5, 6, 2, 0]"""

【教你如何使用Python实现二叉树结构及三种遍历】到此这篇关于教你如何使用Python实现二叉树结构及三种遍历的文章就介绍到这了,更多相关Python实现二叉树结构及三种遍历内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！