mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-06 01:18:02 +00:00
@@ -82,21 +82,59 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": true
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"%run ../bst/bst.py\n",
|
||||
"%load ../bst/bst.py"
|
||||
"# %load ../bst/bst.py\n",
|
||||
"class Node(object):\n",
|
||||
"\n",
|
||||
" def __init__(self, data):\n",
|
||||
" self.data = data\n",
|
||||
" self.left = None\n",
|
||||
" self.right = None\n",
|
||||
" self.parent = None\n",
|
||||
"\n",
|
||||
" def __repr__(self):\n",
|
||||
" return str(self.data)\n",
|
||||
"\n",
|
||||
"\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 TypeError('data cannot be None')\n",
|
||||
" if self.root is None:\n",
|
||||
" self.root = Node(data)\n",
|
||||
" return self.root\n",
|
||||
" else:\n",
|
||||
" return self._insert(self.root, data)\n",
|
||||
"\n",
|
||||
" def _insert(self, node, data):\n",
|
||||
" if node is None:\n",
|
||||
" return Node(data)\n",
|
||||
" if data <= node.data:\n",
|
||||
" if node.left is None:\n",
|
||||
" node.left = self._insert(node.left, data)\n",
|
||||
" node.left.parent = node\n",
|
||||
" return node.left\n",
|
||||
" else:\n",
|
||||
" return self._insert(node.left, data)\n",
|
||||
" else:\n",
|
||||
" if node.right is None:\n",
|
||||
" node.right = self._insert(node.right, data)\n",
|
||||
" node.right.parent = node\n",
|
||||
" return node.right\n",
|
||||
" else:\n",
|
||||
" return self._insert(node.right, data)\n"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"class BstDfs(Bst):\n",
|
||||
@@ -124,9 +162,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": true
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"%run ../utils/results.py"
|
||||
@@ -135,18 +171,17 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"# %load test_dfs.py\n",
|
||||
"from nose.tools import assert_equal\n",
|
||||
"import unittest\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"class TestDfs(object):\n",
|
||||
"class TestDfs(unittest.TestCase):\n",
|
||||
"\n",
|
||||
" def __init__(self):\n",
|
||||
" def __init__(self, *args, **kwargs):\n",
|
||||
" super(TestDfs, self).__init__()\n",
|
||||
" self.results = Results()\n",
|
||||
"\n",
|
||||
" def test_dfs(self):\n",
|
||||
@@ -157,15 +192,15 @@
|
||||
" bst.insert(3)\n",
|
||||
"\n",
|
||||
" bst.in_order_traversal(bst.root, self.results.add_result)\n",
|
||||
" assert_equal(str(self.results), \"[1, 2, 3, 5, 8]\")\n",
|
||||
" self.assertEqual(str(self.results), \"[1, 2, 3, 5, 8]\")\n",
|
||||
" self.results.clear_results()\n",
|
||||
"\n",
|
||||
" bst.pre_order_traversal(bst.root, self.results.add_result)\n",
|
||||
" assert_equal(str(self.results), \"[5, 2, 1, 3, 8]\")\n",
|
||||
" self.assertEqual(str(self.results), \"[5, 2, 1, 3, 8]\")\n",
|
||||
" self.results.clear_results()\n",
|
||||
"\n",
|
||||
" bst.post_order_traversal(bst.root, self.results.add_result)\n",
|
||||
" assert_equal(str(self.results), \"[1, 3, 2, 8, 5]\")\n",
|
||||
" self.assertEqual(str(self.results), \"[1, 3, 2, 8, 5]\")\n",
|
||||
" self.results.clear_results()\n",
|
||||
"\n",
|
||||
" bst = BstDfs(Node(1))\n",
|
||||
@@ -175,15 +210,15 @@
|
||||
" bst.insert(5)\n",
|
||||
"\n",
|
||||
" bst.in_order_traversal(bst.root, self.results.add_result)\n",
|
||||
" assert_equal(str(self.results), \"[1, 2, 3, 4, 5]\")\n",
|
||||
" self.assertEqual(str(self.results), \"[1, 2, 3, 4, 5]\")\n",
|
||||
" self.results.clear_results()\n",
|
||||
"\n",
|
||||
" bst.pre_order_traversal(bst.root, self.results.add_result)\n",
|
||||
" assert_equal(str(self.results), \"[1, 2, 3, 4, 5]\")\n",
|
||||
" self.assertEqual(str(self.results), \"[1, 2, 3, 4, 5]\")\n",
|
||||
" self.results.clear_results()\n",
|
||||
"\n",
|
||||
" bst.post_order_traversal(bst.root, self.results.add_result)\n",
|
||||
" assert_equal(str(self.results), \"[5, 4, 3, 2, 1]\")\n",
|
||||
" self.assertEqual(str(self.results), \"[5, 4, 3, 2, 1]\")\n",
|
||||
"\n",
|
||||
" print('Success: test_dfs')\n",
|
||||
"\n",
|
||||
@@ -223,9 +258,9 @@
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.5.0"
|
||||
"version": "3.7.2"
|
||||
}
|
||||
},
|
||||
"nbformat": 4,
|
||||
"nbformat_minor": 0
|
||||
"nbformat_minor": 1
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user