实现一个二叉搜索树(Binary Search Tree, BST)主要涉及定义树的结构、插入新节点、搜索节点、以及可能的其他操作,如删除节点、遍历树等。下面是一个简单的二叉搜索树的实现示例,使用Python语言:
定义树节点
首先,我们需要定义树的节点,每个节点包含一个值和两个指向其子节点的引用(左子节点和右子节点)。
python复制代码
class TreeNode: |
|
def __init__(self, key): |
|
self.left = None |
|
self.right = None |
|
self.val = key |
定义二叉搜索树
然后,我们可以定义二叉搜索树类,并实现插入、搜索和遍历等方法。
python复制代码
class BinarySearchTree: |
|
def __init__(self): |
|
self.root = None |
|
def insert(self, key): |
|
""" |
|
向二叉搜索树中插入一个新节点 |
|
""" |
|
if self.root is None: |
|
self.root = TreeNode(key) |
|
else: |
|
self._insert_recursive(self.root, key) |
|
def _insert_recursive(self, root, key): |
|
""" |
|
递归地插入新节点 |
|
""" |
|
if key < root.val: |
|
if root.left is None: |
|
root.left = TreeNode(key) |
|
else: |
|
self._insert_recursive(root.left, key) |
|
else: |
|
if root.right is None: |
|
root.right = TreeNode(key) |
|
else: |
|
self._insert_recursive(root.right, key) |
|
def search(self, key): |
|
""" |
|
在二叉搜索树中搜索一个节点 |
|
""" |
|
return self._search_recursive(self.root, key) |
|
def _search_recursive(self, root, key): |
|
""" |
|
递归地搜索节点 |
|
""" |
|
if root is None or root.val == key: |
|
return root |
|
if key < root.val: |
|
return self._search_recursive(root.left, key) |
|
return self._search_recursive(root.right, key) |
|
def inorder_traversal(self): |
|
""" |
|
中序遍历二叉搜索树 |
|
""" |
|
self._inorder_traversal_recursive(self.root) |
|
def _inorder_traversal_recursive(self, root): |
|
""" |
|
递归地执行中序遍历 |
|
""" |
|
if root: |
|
self._inorder_traversal_recursive(root.left) |
|
print(root.val, end=' ') |
|
self._inorder_traversal_recursive(root.right) |
|
# 使用示例 |
|
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.inorder_traversal() # 应该输出 20 30 40 50 60 70 80 |
|
print("搜索 60:", bst.search(60) is not None) # 应该输出 True |
这个简单的二叉搜索树实现包括了插入、搜索和中序遍历功能。你可以根据需要扩展其他功能,如删除节点、计算树的高度、检查树是否平衡等。
注意,二叉搜索树在最坏的情况下(如插入的键已经是有序的)会退化为链表,导致搜索、插入和删除操作的时间复杂度退化到O(n)。为了保持较好的性能,可能需要考虑使用其他平衡二叉树,如AVL树或红黑树。