二叉搜索(排序)树的python实现

二叉搜索树依赖于这样的一个性质:小于父节点的节点都在左子树中,大于父节点的节点都在右子树中。

首先定义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

你可能感兴趣的:(数据结构)