【算法】【python实现】二叉搜索树插入、删除、查找

二叉搜索树

 

定义：如果一颗二叉树的每个节点对应一个关键码值，且关键码值的组织是有顺序的，例如左子节点值小于父节点值，父节点值小于右子节点值，则这棵二叉树是一棵二叉搜索树。

 

类（TreeNode）：定义二叉搜索树各个节点

在该类中，分别存放节点本身的值，以及其左子节点，右子节点，父节点的值。

 

class TreeNode(object):
    def __init__(self,val):
        self.value = val    #存值
        self.left = None    #存本节点的左子节点
        self.right = None   #存本节点的右子节点
        self.father = None  #存本节点的父节点

　　

类（BST）：定义二叉搜索树的各种功能方法

此类用于存放定义二叉树的插入，删除，搜索等方法。

class BST(object):
    def __init__(self,nodeList):
        self.root = None
        for node in nodeList:
            self.bfs_insert(node)

注：self.bfs_insert(node)中的bfs_insert方法在后面实现。放在构造函数中的目的是将一个列表生成一个二叉搜索树列表。

 

方法（bfs_insert）：实现插入功能

 第一步：将根节点（self.root）设置成当前位置节点（cur），即从根节点开始遍历。根节点的父节点（father）初始设置为None。 

 

def bfs_insert(self,node):
    cur = self.root      #设置当前节点为根节点
    father = None        #设置根节点的父节点为None

　　

第二步：查找可以插入的位置

 

1. 当前位置节点（cur）的值与待插入节点（node）的值相等，返回-1（代表无法进行插入操作）。并将父节点（father）值修改为当前位置节点（cur），代表该节点已经遍历完成。

 

if cur.value == node.value:
    return -1
father = cur

　　

2.当前位置节点（cur）的值大于待插入节点（node）的值时，表示待插入节点（node）需继续在当前位置节点的左子树中继续查找。故把当前位置节点（cur）的左节点赋值给当前位置节点（cur），作为下一个要访问的节点对象。

 

if node.value < cur.value:
    cur = cur.left

　　

 

 

 

 

3.当前位置节点（cur）的值小于待插入节点（node）的值时，表示待插入节点（node）需继续在当前位置节点的右子树中继续查找。故把当前位置节点（cur）的右节点赋值给当前位置节点（cur），作为下一个要访问的节点对象。

 

if node.value > cur.value:
    cur = cur.right

　　

 

 

 

 

第三步：找到插入位置后，将其设置为待插入值（node）的父节点

 

node.father = father

　　

第四步：插入操作

 

1.父节点的值为空（即要插入的是个空二叉树），将待插入节点（node）赋值给根节点（root）。

 

if father == None:
    self.root = node

　　

2.待插入节点（node）的值小于父节点（father）的值，将其放到父节点的左子节点

 

if node.value <father.value:
    father.left = node

　

 

 

3.待插入节点（node）的值大于父节点（father）的值，将其放到父节点的右子节点

 

if node.value >father.value:
    father.right = node

 

 

插入功能代码汇总：

def insert(self,node):
    father = None
    cur = self.root

    while cur != None:
         if cur.value == node.value:
         	return -1
         father = cur
         if node.value < cur.value:
         	cur = cur.left
         else:
         	cur = cur.right
    node.father = father
    if father == None:
        self.root = node
    elif node.value < father.value:
        father.left = node
    else:
        father.right = node

　　

 

方法（bfs）：生成二叉搜索树列表

利用队列先进先出的特性，将一个二叉搜索树存放到列表中。

 

def bfs(self):
    if self.root == None:
        return None
    retList = []
    q = queue.Queue()
    q.put(self.root)                
    while q.empty() is not True:
        node = q.get()
        retList.append(node.value)
        if node.left != None:
            q.put(node.left)
        if node.right != None:
            q.put(node.right)
    return retList

 

示例：针对如下二叉搜索树，遍历存放到一个列表过程

 

 

ps：蓝色：二叉树列表 retList    橙色：队列 q

 

方法（bfs_search）：实现查找功能

第一步：将根节点（self.root）设置成当前位置节点（cur），即从根节点开始遍历。

 

def search(self,value):
    cur = self.root

 

如果cur值不为None，执行第二步。否则执行第三步

 

第二步：对比要查找的值（value）与当前位置节点（cur）的值的大小

 

1. 如果当前位置节点（cur）的值等于要查找的值（value），返回当前位置节点（cur）。

 

if cur.value == value:
    return cur

 

 

 

2. 如果当前位置节点（cur）的值小于要查找的值（value），返回当前位置节点（cur）。

 

if cur.value < value:
    cur = cur.right

 

 

 

 

 

 

3. 如果当前位置节点（cur）的值小于要查找的值（value），返回当前位置节点（cur）。

 

if cur.value > value:
    cur = cur.left

 

 

 

 

 

第三步：如果cur的值为None，返回空。即查找的二叉搜索树为空树。

 

return None

 

查找功能代码汇总：

def search(self,value):
    cur = self.root
    while cur != None:
        if cur.value == value:
            return cur
        elif cur.value < value:
            cur = cur.right
        else:
            cur = cur.left
    return None

　　

方法（bfs_delete）：实现删除功能

第一步：将待删除节点（node）的父节点赋值给father变量。

 

def delete(self,node):
    father = node.father

　　

第二步：判断待删除节点（node）的左子树是否为空

 

1. 待删除节点（node）的左子树为空

if node.left == None:

 

a)     待删除节点（node）为根节点

 

将待删除节点（node）的右子节点置为新的根节点（root），且其如果为非空，将其父节点（node.right.father）赋值为空。

 

if father == None:
    self.root = node.right
    if node.right != None:
        node.right.father = None

 

 

 

 

b)     待删除节点（node）为其父节点的左子节点

 

待删除节点（node）的右子节点（node.right）取代其原来的位置，成为其父节点新的左子节点（father.left）。且其右子节点（node.right）不为空，将待删除节点（node）的父节点赋值给它的父节点（node.right.father）

 

if father.left == node:
    father.left = node.right
    if node.right != None:
        node.right.father = father

 

　

 

 

c)     待删除节点（node）为其父节点的右子节点

 

待删除节点（node）的右子节点（node.right）取代其原来的位置，成为其父节点新的右子节点（father.right）。且其右子节点（node.right）不为空，将待删除节点（node）的父节点赋值给它的父节点（node.right.father）

 

if father.right == node:
    father.right = node.right
    if node.right != None:
        node.right.father = father

 

 

 

2. 待删除节点（node）的左子树不为空

 

第一步：将待删除节点（node）的右子树挂到其左子树最后一层的右子节点下

 

a)     将待删除节点的左子节点（node.left）存到临时变量tmpNode中

 

tmpNode = node.left

 

 

b)     递归找到node.left的最后一层右子节点

 

while tmpNode.right != None:
    tmpNode = tmpNode.right

 

 

c)     将待删节点的右子树（node.right）挂到node.left的最后一层右子节点下

 

tmpNode.right = node.right

 

d)     将node.right的父节点设置为待删节点左子树中的最后一层的右子节点

 

if node.right != None:
    node.right.father = tmpNode

 

　

 

第二步：开始删除node

 

1.待删除节点为根节点

 

将待删除节点的左子节点（node.left）设置为根节点（self.root），并将其父节点设置为空。

 

if father == None:
    self.root = node.left
    node.left.father = None

　　

 

 

 

2.待删除节点为根节点的左子节点

 

待删除节点的左子节点（node.left）取代待删节点的位置，并将其父节点设置为待删除节点的父节点。

 

if father.left == node:
    father.left = node.left
    node.left.father = father

 

 

 

3. 待删除节点为根节点的右子节点

 

待删除节点的左子节点（node.left）取代待删节点的位置，并将其父节点设置为待删除节点的父节点

 

if father.right == node:
    father.right = node.left
    node.left.father = father

 

 

 

删除功能代码汇总：

def delete(self,node):
    father = node.father
    if node.left == None:
        if father == None:
            self.root = node.right
            if node.right != None:
                node.right.father = None
        elif father.left == node:
            father.left = node.right
            if node.right != None:
                node.right.father = father
        else:
            father.right = node.right
            if node.right != None:
                node.right.father = father
        return 'delete successfully'
    tmpNode = node.left
    while tmpNode.right != None:
        tmpNode = tmpNode.right

    tmpNode.right = node.right
    if node.right != None:
        node.right.father = tmpNode

    if father == None:
        self.root = node.left
        node.left.father = None
    elif father.left == node:
        father.left = node.left
        node.left.father = father
    else:
        father.right = node.left
        node.left.father = father
    node = None
    return 'delete successfully'

　

综上，二叉搜索树代码

# encoding=utf-8
import queue

class TreeNode(object):
    def __init__(self,val):
        self.value = val
        self.left = None
        self.right = None
        self.father = None

class BST(object):
    def __init__(self,nodeList):
        self.root = None
        for node in nodeList:
            self.bfs_insert(node)

    def bfs_insert(self,node):
        father = None
        cur = self.root

        while cur != None:
            if cur.value == node.value:
                return -1
            father = cur
            if node.value < cur.value:
                cur = cur.left
            else:
                cur = cur.right
        node.father = father
        if father == None:
            self.root = node
        elif node.value < father.value:
            father.left = node
        else:
            father.right = node

    def bfs(self):
        if self.root == None:
            return None
        retList = []
        q = queue.Queue()
        q.put(self.root)
        while q.empty() is not True:
            node = q.get()
            retList.append(node.value)
            if node.left != None:
                q.put(node.left)
            if node.right != None:
                q.put(node.right)
        return retList

    def bfs_search(self,value):
        cur = self.root
        while cur != None:
            if cur.value == value:
                return cur
            elif cur.value < value:
                cur = cur.right
            else:
                cur = cur.left
        return None

    def bfs_delete(self,node):
        father = node.father
        if node.left == None:
            if father == None:
                self.root = node.right
                if node.right != None:
                    node.right.father = None
            elif father.left == node:
                father.left = node.right
                if node.right != None:
                    node.right.father = father
            else:
                father.right = node.right
                if node.right != None:
                    node.right.father = father
            return 'delete successfully'
        tmpNode = node.left
        while tmpNode.right != None:
            tmpNode = tmpNode.right

        tmpNode.right = node.right
        if node.right != None:
            node.right.father = tmpNode

        if father == None:
            self.root = node.left
            node.left.father = None
        elif father.left == node:
            father.left = node.left
            node.left.father = father
        else:
            father.right = node.left
            node.left.father = father
        node = None
        return 'delete successfully'

if __name__ == '__main__':
    varList = [24,34,5,4,8,23,45,35,28,6,29]
    nodeList = [TreeNode(var) for var in varList]
    bst = BST(nodeList)
    print (bst.bfs())
    node = bst.bfs_search(34)
    bst.bfs_delete(node)
    print (bst.bfs())