iframe-proxy | Sunbelt Computer Software

History

53 lines (43 loc) · 1.1 KB

Raw

"""

Tree_sort algorithm.

Build a BST and in order traverse.

"""

class node:

# BST data structure

def __init__(self, val):

self.val = val

self.left = None

self.right = None

def insert(self, val):

if self.val:

if val < self.val:

if self.left is None:

self.left = node(val)

else:

self.left.insert(val)

elif val > self.val:

if self.right is None:

self.right = node(val)

else:

self.right.insert(val)

else:

self.val = val

def inorder(root, res):

# Recursive traversal

if root:

inorder(root.left, res)

res.append(root.val)

inorder(root.right, res)

def tree_sort(arr):

# Build BST

if len(arr) == 0:

return arr

root = node(arr[0])

for i in range(1, len(arr)):

root.insert(arr[i])

# Traverse BST in order.

res = []

inorder(root, res)

return res

if __name__ == "__main__":

print(tree_sort([10, 1, 3, 2, 9, 14, 13]))

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

tree_sort.py

Latest commit

History

tree_sort.py

File metadata and controls