StarMirror/interactive-coding-challenges
1
0
Fork 0
mirror of https://github.com/donnemartin/interactive-coding-challenges.git synced 2024-03-22 13:11:13 +08:00
Code Issues Projects Releases Wiki Activity

Move bst to a class

Browse Source
This commit is contained in:
Donne Martin 2016-08-14 08:20:06 -04:00
parent be5f29f98e
commit 82c9725000
4 changed files with 119 additions and 81 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
graphs_trees/bst/bst.py
View File

@ -10,22 +10,34 @@ class Node(object):
return str(self.data) return str(self.data)
def insert(root, data): class Bst(object):
def __init__(self, root=None):
self.root = root
def insert(self, data):
if data is None:
raise Exception('Data cannot be None')
if self.root is None:
self.root = Node(data)
return self.root
return self._insert(self.root, data)
def _insert(self, node, data):
# Constraint: Assume we are working with valid ints # Constraint: Assume we are working with valid ints
if root is None: if node is None:
root = Node(data) return Node(data)
return root if data <= node.data:
if data <= root.data: if node.left is None:
if root.left is None: node.left = self._insert(node.left, data)
root.left = insert(root.left, data) node.left.parent = node
root.left.parent = root return node.left
return root.left
else: else:
return insert(root.left, data) return self._insert(node.left, data)
else: else:
if root.right is None: if node.right is None:
root.right = insert(root.right, data) node.right = self._insert(node.right, data)
root.right.parent = root node.right.parent = node
return root.right return node.right
else: else:
return insert(root.right, data) return self._insert(node.right, data)

36
graphs_trees/bst/bst_challenge.ipynb
View File

@ -98,7 +98,9 @@
" pass\n", " pass\n",
"\n", "\n",
"\n", "\n",
"def insert(root, data):\n", "class Bst(object):\n",
"\n",
" def insert(self, data):\n",
" # TODO: Implement me\n", " # TODO: Implement me\n",
" pass" " pass"
] ]
@ -156,22 +158,25 @@
" def __init__(self):\n", " def __init__(self):\n",
" self.results = Results()\n", " self.results = Results()\n",
"\n", "\n",
" def test_tree(self):\n", " def test_tree_one(self):\n",
" node = Node(5)\n", " bst = Bst()\n",
" assert_equal(insert(node, 2).data, 2)\n", " bst.insert(5)\n",
" assert_equal(insert(node, 8).data, 8)\n", " bst.insert(2)\n",
" assert_equal(insert(node, 1).data, 1)\n", " bst.insert(8)\n",
" assert_equal(insert(node, 3).data, 3)\n", " bst.insert(1)\n",
" in_order_traversal(node, self.results.add_result)\n", " bst.insert(3)\n",
" in_order_traversal(bst.root, self.results.add_result)\n",
" assert_equal(str(self.results), '[1, 2, 3, 5, 8]')\n", " assert_equal(str(self.results), '[1, 2, 3, 5, 8]')\n",
" self.results.clear_results()\n", " self.results.clear_results()\n",
"\n", "\n",
" node = insert(None, 1)\n", " def test_tree_two(self):\n",
" assert_equal(insert(node, 2).data, 2)\n", " bst = Bst()\n",
" assert_equal(insert(node, 3).data, 3)\n", " bst.insert(1)\n",
" assert_equal(insert(node, 4).data, 4)\n", " bst.insert(2)\n",
" insert(node, 5)\n", " bst.insert(3)\n",
" in_order_traversal(node, self.results.add_result)\n", " bst.insert(4)\n",
" bst.insert(5)\n",
" in_order_traversal(bst.root, self.results.add_result)\n",
" assert_equal(str(self.results), '[1, 2, 3, 4, 5]')\n", " assert_equal(str(self.results), '[1, 2, 3, 4, 5]')\n",
"\n", "\n",
" print('Success: test_tree')\n", " print('Success: test_tree')\n",
@ -179,7 +184,8 @@
"\n", "\n",
"def main():\n", "def main():\n",
" test = TestTree()\n", " test = TestTree()\n",
" test.test_tree()\n", " test.test_tree_one()\n",
" test.test_tree_two()\n",
"\n", "\n",
"\n", "\n",
"if __name__ == '__main__':\n", "if __name__ == '__main__':\n",

74
graphs_trees/bst/bst_solution.ipynb
View File

@ -127,25 +127,37 @@
" return str(self.data)\n", " return str(self.data)\n",
"\n", "\n",
"\n", "\n",
"def insert(root, data):\n", "class Bst(object):\n",
"\n",
" def __init__(self, root=None):\n",
" self.root = root\n",
"\n",
" def insert(self, data):\n",
" if data is None:\n",
" raise Exception('Data cannot be None')\n",
" if self.root is None:\n",
" self.root = Node(data)\n",
" return self.root\n",
" return self._insert(self.root, data)\n",
"\n",
" def _insert(self, node, data):\n",
" # Constraint: Assume we are working with valid ints\n", " # Constraint: Assume we are working with valid ints\n",
" if root is None:\n", " if node is None:\n",
" root = Node(data)\n", " return Node(data)\n",
" return root\n", " if data <= node.data:\n",
" if data <= root.data:\n", " if node.left is None:\n",
" if root.left is None:\n", " node.left = self._insert(node.left, data)\n",
" root.left = insert(root.left, data)\n", " node.left.parent = node\n",
" root.left.parent = root\n", " return node.left\n",
" return root.left\n",
" else:\n", " else:\n",
" return insert(root.left, data)\n", " return self._insert(node.left, data)\n",
" else:\n", " else:\n",
" if root.right is None:\n", " if node.right is None:\n",
" root.right = insert(root.right, data)\n", " node.right = self._insert(node.right, data)\n",
" root.right.parent = root\n", " node.right.parent = node\n",
" return root.right\n", " return node.right\n",
" else:\n", " else:\n",
" return insert(root.right, data)" " return self._insert(node.right, data)"
] ]
}, },
{ {
@ -213,22 +225,25 @@
" def __init__(self):\n", " def __init__(self):\n",
" self.results = Results()\n", " self.results = Results()\n",
"\n", "\n",
" def test_tree(self):\n", " def test_tree_one(self):\n",
" node = Node(5)\n", " bst = Bst()\n",
" assert_equal(insert(node, 2).data, 2)\n", " bst.insert(5)\n",
" assert_equal(insert(node, 8).data, 8)\n", " bst.insert(2)\n",
" assert_equal(insert(node, 1).data, 1)\n", " bst.insert(8)\n",
" assert_equal(insert(node, 3).data, 3)\n", " bst.insert(1)\n",
" in_order_traversal(node, self.results.add_result)\n", " bst.insert(3)\n",
" in_order_traversal(bst.root, self.results.add_result)\n",
" assert_equal(str(self.results), '[1, 2, 3, 5, 8]')\n", " assert_equal(str(self.results), '[1, 2, 3, 5, 8]')\n",
" self.results.clear_results()\n", " self.results.clear_results()\n",
"\n", "\n",
" node = insert(None, 1)\n", " def test_tree_two(self):\n",
" assert_equal(insert(node, 2).data, 2)\n", " bst = Bst()\n",
" assert_equal(insert(node, 3).data, 3)\n", " bst.insert(1)\n",
" assert_equal(insert(node, 4).data, 4)\n", " bst.insert(2)\n",
" insert(node, 5)\n", " bst.insert(3)\n",
" in_order_traversal(node, self.results.add_result)\n", " bst.insert(4)\n",
" bst.insert(5)\n",
" in_order_traversal(bst.root, self.results.add_result)\n",
" assert_equal(str(self.results), '[1, 2, 3, 4, 5]')\n", " assert_equal(str(self.results), '[1, 2, 3, 4, 5]')\n",
"\n", "\n",
" print('Success: test_tree')\n", " print('Success: test_tree')\n",
@ -236,7 +251,8 @@
"\n", "\n",
"def main():\n", "def main():\n",
" test = TestTree()\n", " test = TestTree()\n",
" test.test_tree()\n", " test.test_tree_one()\n",
" test.test_tree_two()\n",
"\n", "\n",
"\n", "\n",
"if __name__ == '__main__':\n", "if __name__ == '__main__':\n",

32
graphs_trees/bst/test_bst.py
View File

@ -6,22 +6,25 @@ class TestTree(object):
def __init__(self): def __init__(self):
self.results = Results() self.results = Results()
def test_tree(self): def test_tree_one(self):
node = Node(5) bst = Bst()
assert_equal(insert(node, 2).data, 2) bst.insert(5)
assert_equal(insert(node, 8).data, 8) bst.insert(2)
assert_equal(insert(node, 1).data, 1) bst.insert(8)
assert_equal(insert(node, 3).data, 3) bst.insert(1)
in_order_traversal(node, self.results.add_result) bst.insert(3)
in_order_traversal(bst.root, self.results.add_result)
assert_equal(str(self.results), '[1, 2, 3, 5, 8]') assert_equal(str(self.results), '[1, 2, 3, 5, 8]')
self.results.clear_results() self.results.clear_results()
node = insert(None, 1) def test_tree_two(self):
assert_equal(insert(node, 2).data, 2) bst = Bst()
assert_equal(insert(node, 3).data, 3) bst.insert(1)
assert_equal(insert(node, 4).data, 4) bst.insert(2)
insert(node, 5) bst.insert(3)
in_order_traversal(node, self.results.add_result) bst.insert(4)
bst.insert(5)
in_order_traversal(bst.root, self.results.add_result)
assert_equal(str(self.results), '[1, 2, 3, 4, 5]') assert_equal(str(self.results), '[1, 2, 3, 4, 5]')
print('Success: test_tree') print('Success: test_tree')
@ -29,7 +32,8 @@ class TestTree(object):
def main(): def main():
test = TestTree() test = TestTree()
test.test_tree() test.test_tree_one()
test.test_tree_two()
if __name__ == '__main__': if __name__ == '__main__':