Browse Source

Forget camel case

master
Raphael Roberts 6 years ago
parent
commit
eb5836ce40
  1. 14
      BST.py

14
BST.py

@ -46,9 +46,9 @@ class BST:
parent.left = to_add parent.left = to_add
self.size += 1 self.size += 1
def inOrder(self):
def in_order(self):
ret = [] ret = []
inOrderRec(self.root, ret)
in_order_rec(self.root, ret)
return ret return ret
@classmethod @classmethod
@ -61,14 +61,14 @@ class BST:
return self.root.height() return self.root.height()
def inOrderRec(top, ret):
def in_order_rec(top, ret):
if top is None: if top is None:
return return
if top.left: if top.left:
inOrderRec(top.left, ret)
in_order_rec(top.left, ret)
ret.append(top) ret.append(top)
if top.right: if top.right:
inOrderRec(top.right, ret)
in_order_rec(top.right, ret)
def add_rec(start, end, list_param, tree: BST): def add_rec(start, end, list_param, tree: BST):
@ -86,6 +86,6 @@ if __name__ == "__main__":
tree2 = BST() tree2 = BST()
for i in range(test): for i in range(test):
tree2.add(i) tree2.add(i)
r = list(node.value for node in tree.inOrder())
r2 = list(node.value for node in tree2.inOrder())
r = list(node.value for node in tree.in_order())
r2 = list(node.value for node in tree2.in_order())
assert r == r2 assert r == r2
Loading…
Cancel
Save