快捷導(dǎo)航

python如何實(shí)現(xiàn)二叉搜索樹算法

更新時(shí)間：2024年10月21日 09:14:49 作者：luthane

二叉搜索樹（BST）是一種數(shù)據(jù)結(jié)構(gòu),用于動(dòng)態(tài)集合操作如搜索、插入、刪除等,每個(gè)節(jié)點(diǎn)的左子樹包含小于節(jié)點(diǎn)值的所有項(xiàng),右子樹包含大于節(jié)點(diǎn)值的所有項(xiàng),通過(guò)中序遍歷可得升序序列,插入、搜索和刪除都從根節(jié)點(diǎn)開始,根據(jù)值的大小移動(dòng)到左或右子樹

二叉搜索樹算法介紹

二叉搜索樹（Binary Search Tree，簡(jiǎn)稱BST）是一種常見(jiàn)的數(shù)據(jù)結(jié)構(gòu)，它支持一系列的動(dòng)態(tài)集合操作，包括搜索、插入、刪除和遍歷等。

在二叉搜索樹中，對(duì)于樹中的每個(gè)節(jié)點(diǎn)X，其左子樹中的所有項(xiàng)的值都小于X中的項(xiàng)，而其右子樹中的所有項(xiàng)的值都大于X中的項(xiàng)。

1. 二叉搜索樹的性質(zhì)

唯一根節(jié)點(diǎn)：非空二叉搜索樹有一個(gè)根節(jié)點(diǎn)。
左子樹：對(duì)于樹中的每個(gè)節(jié)點(diǎn)X，其左子樹中的所有項(xiàng)的值都小于X中的項(xiàng)。
右子樹：對(duì)于樹中的每個(gè)節(jié)點(diǎn)X，其右子樹中的所有項(xiàng)的值都大于X中的項(xiàng)。
中序遍歷：對(duì)二叉搜索樹進(jìn)行中序遍歷（左-根-右）可以得到一個(gè)按升序排列的節(jié)點(diǎn)值的序列。

2. 基本操作

插入

從根節(jié)點(diǎn)開始。
如果要插入的值小于當(dāng)前節(jié)點(diǎn)的值，移動(dòng)到左子樹。
如果要插入的值大于當(dāng)前節(jié)點(diǎn)的值，移動(dòng)到右子樹。
重復(fù)步驟2和3，直到找到一個(gè)空位置插入新節(jié)點(diǎn)。

搜索

從根節(jié)點(diǎn)開始。
如果要搜索的值小于當(dāng)前節(jié)點(diǎn)的值，移動(dòng)到左子樹。
如果要搜索的值大于當(dāng)前節(jié)點(diǎn)的值，移動(dòng)到右子樹。
重復(fù)步驟2和3，直到找到值相等或到達(dá)葉子節(jié)點(diǎn)（無(wú)子節(jié)點(diǎn)）。

刪除

刪除操作稍微復(fù)雜一些，因?yàn)樗枰幚砣N情況：

要?jiǎng)h除的節(jié)點(diǎn)是葉子節(jié)點(diǎn)：直接刪除該節(jié)點(diǎn)，并修改其父節(jié)點(diǎn)的鏈接。
要?jiǎng)h除的節(jié)點(diǎn)有一個(gè)子節(jié)點(diǎn)：用其子節(jié)點(diǎn)替換該節(jié)點(diǎn)，并修改其父節(jié)點(diǎn)的鏈接。
要?jiǎng)h除的節(jié)點(diǎn)有兩個(gè)子節(jié)點(diǎn)：找到該節(jié)點(diǎn)的右子樹中的最小節(jié)點(diǎn)（或左子樹中的最大節(jié)點(diǎn)），用該節(jié)點(diǎn)的值替換要?jiǎng)h除的節(jié)點(diǎn)的值，并刪除右子樹中的最小節(jié)點(diǎn)（或左子樹中的最大節(jié)點(diǎn)）。

3. 示例代碼（Python）

這里僅提供一個(gè)非?；镜牟迦牒退阉鞯氖纠a：

class TreeNode:
    def __init__(self, key):
        self.left = None
        self.right = None
        self.val = key

class BinarySearchTree:
    def __init__(self):
        self.root = None

    def insert(self, key):
        if self.root is None:
            self.root = TreeNode(key)
        else:
            self._insert_rec(self.root, key)

    def _insert_rec(self, root, key):
        if key < root.val:
            if root.left is None:
                root.left = TreeNode(key)
            else:
                self._insert_rec(root.left, key)
        elif key > root.val:
            if root.right is None:
                root.right = TreeNode(key)
            else:
                self._insert_rec(root.right, key)

    def search(self, key):
        return self._search_rec(self.root, key)

    def _search_rec(self, root, key):
        if root is None or root.val == key:
            return root is not None
        if key < root.val:
            return self._search_rec(root.left, key)
        return self._search_rec(root.right, key)

# 使用示例
bst = BinarySearchTree()
bst.insert(50)
bst.insert(30)
bst.insert(20)
bst.insert(40)
bst.insert(70)
bst.insert(60)
bst.insert(80)

print(bst.search(40))  # 輸出: True
print(bst.search(25))  # 輸出: False

請(qǐng)注意：

這個(gè)示例代碼只實(shí)現(xiàn)了插入和搜索功能，并沒(méi)有實(shí)現(xiàn)刪除操作。
刪除操作需要更多的邏輯來(lái)處理不同的情況。

二叉搜索樹算法python實(shí)現(xiàn)樣例

下面是一個(gè)簡(jiǎn)單的 Python 實(shí)現(xiàn)二叉搜索樹的算法：

# 定義二叉搜索樹的節(jié)點(diǎn)
class Node:
    def __init__(self, value):
        self.value = value
        self.left = None
        self.right = None

# 定義二叉搜索樹
class BinarySearchTree:
    def __init__(self):
        self.root = None

    # 插入節(jié)點(diǎn)
    def insert(self, value):
        if self.root is None:
            self.root = Node(value)
        else:
            self._insert_recursive(self.root, value)

    def _insert_recursive(self, node, value):
        if value < node.value:
            if node.left is None:
                node.left = Node(value)
            else:
                self._insert_recursive(node.left, value)
        else:
            if node.right is None:
                node.right = Node(value)
            else:
                self._insert_recursive(node.right, value)

    # 查找節(jié)點(diǎn)
    def find(self, value):
        return self._find_recursive(self.root, value)

    def _find_recursive(self, node, value):
        if node is None or node.value == value:
            return node
        if value < node.value:
            return self._find_recursive(node.left, value)
        return self._find_recursive(node.right, value)

    # 刪除節(jié)點(diǎn)
    def delete(self, value):
        self.root = self._delete_recursive(self.root, value)

    def _delete_recursive(self, node, value):
        if node is None:
            return node
        if value < node.value:
            node.left = self._delete_recursive(node.left, value)
        elif value > node.value:
            node.right = self._delete_recursive(node.right, value)
        else:
            # 找到要?jiǎng)h除的節(jié)點(diǎn)
            if node.left is None:
                return node.right
            elif node.right is None:
                return node.left
            else:
                # 找到右子樹中的最小節(jié)點(diǎn)，替換當(dāng)前節(jié)點(diǎn)
                min_node = self._find_min(node.right)
                node.value = min_node.value
                node.right = self._delete_recursive(node.right, min_node.value)
        return node

    def _find_min(self, node):
        while node.left is not None:
            node = node.left
        return node

    # 中序遍歷
    def inorder_traversal(self):
        self._inorder_traversal_recursive(self.root)

    def _inorder_traversal_recursive(self, node):
        if node is not None:
            self._inorder_traversal_recursive(node.left)
            print(node.value, end=" ")
            self._inorder_traversal_recursive(node.right)

使用方法：

bst = BinarySearchTree()
bst.insert(8)
bst.insert(3)
bst.insert(10)
bst.insert(1)
bst.insert(6)
bst.insert(14)
bst.insert(4)
bst.insert(7)
bst.insert(13)

bst.inorder_traversal()  # 輸出：1 3 4 6 7 8 10 13 14

bst.delete(8)
bst.inorder_traversal()  # 輸出：1 3 4 6 7 10 13 14

node = bst.find(6)
print(node.value)  # 輸出：6

這是一個(gè)基本的二叉搜索樹算法實(shí)現(xiàn)，包括插入、查找、刪除和中序遍歷操作。你可以根據(jù)需要進(jìn)一步擴(kuò)展和優(yōu)化這個(gè)實(shí)現(xiàn)。