Viết lại hàm BSTSort(A) thực hiện sắp xếp dãy số A theo thứ tự tăng dần nhưng kết quả không cập nhật vào A. Hàm trả lại dãy số mới là dãy vừa được sắp xếp (gồm các phần tử của dãy A).
1. Định nghĩa cấu trúc nút cây BST:
class TreeNode:
def __init__(self, key):
self.left = None
self.right = None
self.val = key
2. Hàm chèn một phần tử vào BST
def insert(root, key):
if root is None:
return TreeNode(key)
else:
if root.val < key:
root.right = insert(root.right, key)
else:
root.left = insert(root.left, key)
return root
3. Hàm duyệt giữa để lấy thứ tự tăng dần
def in_order_traversal(root, result):
if root:
in_order_traversal(root.left, result)
result.append(root.val)
in_order_traversal(root.right, result)
4. Hàm chính để sắp xếp dãy A
def BSTSort(A):
if not A:
return []
# Bước 1: Tạo cây tìm kiếm nhị phân từ dãy A
root = None
for key in A:
root = insert(root, key)
# Bước 2: Duyệt cây để lấy kết quả sắp xếp
sorted_result = []
in_order_traversal(root, sorted_result)
return sorted_result