如何使用python实现二叉排序树

如何使用python实现二叉排序树

小编给大家分享一下如何使用python实现二叉排序树，希望大家阅读完这篇文章之后都有所收获，下面让我们一起去探讨吧！

方法一（粗暴）

#二叉排序树classBTree():def__init__(self,data):self.left=Noneself.right=Noneiftype(data)==list:self.data=data[0]fordindata[1:]:self.insert(d)else:self.data=datadefinsert(self,data):bt=selfwhileTrue:ifdata<=bt.data:ifbt.left==None:bt.left=BTree(data)breakelse:bt=bt.leftelse:ifbt.right==None:bt.right=BTree(data)breakelse:bt=bt.rightdefmid_order(self):res=[]stack=[]node=selfwhilenodeorstack:whilenode:stack.append(node)node=node.leftnode=stack.pop()res.append(node.data)node=node.rightreturnresdata=[5,1,2,3,6,8,9]bt=BTree(data)print(bt.mid_order())

方法二（递归）

classTreeNode(object):def__init__(self,data):self.data=dataself.left=Noneself.right=NoneclassBinaryTree(object):definsert(self,root,node):ifrootisNone:returnnodeifnode.data<root.data:root.left=self.insert(root.left,node)else:root.right=self.insert(root.right,node)returnrootdefmid_order(self,root):node=rootstack=[]res=[]whilenodeorstack:whilenode:stack.append(node)node=node.leftnode=stack.pop()res.append(node.data)node=node.rightreturnresdata=[5,1,2,3,6,8,9]root=TreeNode(data[0])tree=BinaryTree()foriindata[1:]:tree.insert(root,TreeNode(i))print(tree.mid_order(root))

看完了这篇文章，相信你对“如何使用python实现二叉排序树”有了一定的了解，如果想了解更多相关知识，欢迎关注恰卡编程网行业资讯频道，感谢各位的阅读！