interactive-coding-challenges/graphs_trees/bst/bst.py

27 lines
648 B
Python
Raw Normal View History

2015-08-02 05:44:14 +08:00
class Node(object):
def __init__(self, data):
self.data = data
self.left = None
self.right = None
self.parent = None
2015-08-02 05:44:14 +08:00
def __str__(self):
return str(self.data)
2015-08-02 05:44:14 +08:00
def insert(root, data):
if data <= root.data:
if root.left is None:
root.left = Node(data)
root.left.parent = root
return root.left
2015-08-02 05:44:14 +08:00
else:
return insert(root.left, data)
2015-08-02 05:44:14 +08:00
else:
if root.right is None:
root.right = Node(data)
root.right.parent = root
return root.right
2015-08-02 05:44:14 +08:00
else:
return insert(root.right, data)