二叉搜索树依赖于这样的一个性质:小于父节点的节点都在左子树中,大于父节点的节点都在右子树中。
首先定义BST(BinarySearchTree)类和Node类:
class Node:
def __init__(self,val):
self.val=val
self.lt=None
self.rt=None
self.parent=None
class BST:
def __init__(self,node_list=None):
if node_list==None:
self.root=None
else:
self.root=Node(node_list[0])
for i in node_list:
self.insert(i)
初始化BST类时调用了insert函数,当插入一个节点到BST中时,首先比较插入节点与根节点的大小,如果插入节点小于根节点,则比较插入节点与左孩子大小,如果插入节点大于根节点,则比较插入节点与右孩子大小,一直循环下去直到寻找到插入位置。
非递归版:
def insert(self,val):
if not self.root:
self.root=Node(val)
else:
new=Node(val)
temp=self.root
while temp:
if val<=temp.val:
cur=temp
temp=temp.lt
else:
cur=temp
temp=temp.rt
new.parent=cur
if val<=cur.val:
cur.lt=new
else:
cur.rt=new
当搜索某个元素是否在BST中时,用到search函数。和insert函数一样,首先从根节点搜索,如果目标元素小于根节点,则查询左子树,否则查询右子树,直到搜索到目标元素。
非递归版:
def search(self,obj):
cur=self.rt
while cur:
if cur.val==obj:
print('Find!')
return
if obj<cur.val:
cur=cur.lt
else:
cur=cur.rt
if not cur:
print('No Find!')