mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-05 00:48:03 +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
|
||||
}
|
||||
|
||||
@@ -116,9 +116,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 1,
|
||||
"metadata": {
|
||||
"collapsed": true
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"%run ../bst/bst.py"
|
||||
@@ -127,9 +125,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 2,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"class BstDfs(Bst):\n",
|
||||
@@ -163,9 +159,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 3,
|
||||
"metadata": {
|
||||
"collapsed": true
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"%run ../utils/results.py"
|
||||
@@ -174,9 +168,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 4,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
@@ -188,12 +180,13 @@
|
||||
],
|
||||
"source": [
|
||||
"%%writefile 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",
|
||||
@@ -204,15 +197,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",
|
||||
@@ -222,15 +215,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",
|
||||
@@ -247,9 +240,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 5,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
@@ -280,9 +271,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
|
||||
}
|
||||
|
||||
@@ -1,9 +1,10 @@
|
||||
from nose.tools import assert_equal
|
||||
import unittest
|
||||
|
||||
|
||||
class TestDfs(object):
|
||||
class TestDfs(unittest.TestCase):
|
||||
|
||||
def __init__(self):
|
||||
def __init__(self, *args, **kwargs):
|
||||
super(TestDfs, self).__init__()
|
||||
self.results = Results()
|
||||
|
||||
def test_dfs(self):
|
||||
@@ -14,15 +15,15 @@ class TestDfs(object):
|
||||
bst.insert(3)
|
||||
|
||||
bst.in_order_traversal(bst.root, self.results.add_result)
|
||||
assert_equal(str(self.results), "[1, 2, 3, 5, 8]")
|
||||
self.assertEqual(str(self.results), "[1, 2, 3, 5, 8]")
|
||||
self.results.clear_results()
|
||||
|
||||
bst.pre_order_traversal(bst.root, self.results.add_result)
|
||||
assert_equal(str(self.results), "[5, 2, 1, 3, 8]")
|
||||
self.assertEqual(str(self.results), "[5, 2, 1, 3, 8]")
|
||||
self.results.clear_results()
|
||||
|
||||
bst.post_order_traversal(bst.root, self.results.add_result)
|
||||
assert_equal(str(self.results), "[1, 3, 2, 8, 5]")
|
||||
self.assertEqual(str(self.results), "[1, 3, 2, 8, 5]")
|
||||
self.results.clear_results()
|
||||
|
||||
bst = BstDfs(Node(1))
|
||||
@@ -32,15 +33,15 @@ class TestDfs(object):
|
||||
bst.insert(5)
|
||||
|
||||
bst.in_order_traversal(bst.root, self.results.add_result)
|
||||
assert_equal(str(self.results), "[1, 2, 3, 4, 5]")
|
||||
self.assertEqual(str(self.results), "[1, 2, 3, 4, 5]")
|
||||
self.results.clear_results()
|
||||
|
||||
bst.pre_order_traversal(bst.root, self.results.add_result)
|
||||
assert_equal(str(self.results), "[1, 2, 3, 4, 5]")
|
||||
self.assertEqual(str(self.results), "[1, 2, 3, 4, 5]")
|
||||
self.results.clear_results()
|
||||
|
||||
bst.post_order_traversal(bst.root, self.results.add_result)
|
||||
assert_equal(str(self.results), "[5, 4, 3, 2, 1]")
|
||||
self.assertEqual(str(self.results), "[5, 4, 3, 2, 1]")
|
||||
|
||||
print('Success: test_dfs')
|
||||
|
||||
@@ -51,4 +52,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user