树排序是一种在线排序算法。它使用二叉搜索树数据结构来存储元素。通过按顺序遍历二叉搜索树,可以按排序顺序检索元素。由于它是一种在线排序算法,因此插入的元素始终按排序顺序进行维护。
假设使用一组未排序的数组 array
包含 n
个元素。
算法主体的步骤:
- 通过在二叉搜索树中插入数组中的元素来构建二进制搜索树;
- 在树上执行顺序遍历,以使元素按排序顺序返回。
插入排序的步骤:
- 创建一个BST节点,其值等于数组元素
array[i]
; - Insert(node, key): 如果
root
== null,那么返回新形成的节点;如果root->data
<key
,那么root->right
=insert(root->right, key)
;如果root->data
>key
,那么root->left
=insert(root->left, key)
; - 返回指向原始根结点的游标。
顺序遍历操作:遍历左子树 → 访问根结点 → 遍历右子树。
在平均情况下,在BST中插入n个节点的时间复杂度为
最坏的情况发生在数组排序时,并形成最大高度为
最佳情况发生在形成的二叉搜索树平衡时。时间复杂度的最佳情况是
该算法的空间复杂度为:$O(n)$,因为必须为二叉搜索树中的每个元素创建
class node():
def __init__(self, val): # BST data structure
self.val = val
self.left = None
self.right = None
def insert(self, val) -> None:
if self.val:
if val < self.val: # 若逆序此处可以变为 val > self.val
if self.left is None:
self.left = node(val)
else:
self.left.insert(val)
else:
if self.right is None:
self.right = node(val)
else:
self.right.insert(val)
else:
self.val = val
def inorder(root: node, result: list) -> None:
'''
root: 存储插入序列的根节点指示。
result: 存储遍历结果的数组.
'''
if root:
inorder(root.left, result)
result.append(root.val)
inorder(root.right, result)
def tree_sort(array: list, reverse: bool=False) -> list:
'''
array: 支持数值型数据,如整型与浮点型混合;支持全为字符串类型的数据;不支持字符串型与数值型混合。
reverse: 是否降序, 默认采用升序。
'''
if not array:
return array
root = node(array[0]) # Initial
for index in range(1, len(array)): # Build
root.insert(array[index])
result = []
inorder(root, result) # Traverse BST in order
if reverse:
result.reverse()
return result