2015-08-02 22:53:35 +08:00
{
" cells " : [
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
2016-07-31 20:11:18 +08:00
" This notebook was prepared by [Donne Martin](https://github.com/donnemartin). Source and license info is on [GitHub](https://github.com/donnemartin/interactive-coding-challenges). "
2015-08-02 22:53:35 +08:00
]
} ,
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
" # Solution Notebook "
]
} ,
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
" ## Problem: Determine the height of a tree. \n " ,
" \n " ,
" * [Constraints](#Constraints) \n " ,
" * [Test Cases](#Test-Cases) \n " ,
" * [Algorithm](#Algorithm) \n " ,
" * [Code](#Code) \n " ,
" * [Unit Test](#Unit-Test) "
]
} ,
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
" ## Constraints \n " ,
" \n " ,
" * Is this a binary tree? \n " ,
" * Yes \n " ,
" * Can we assume we already have a Node class with an insert method? \n " ,
2016-06-28 16:33:10 +08:00
" * Yes \n " ,
" * Can we assume this fits memory? \n " ,
2015-08-02 22:53:35 +08:00
" * Yes "
]
} ,
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
" ## Test Cases \n " ,
" \n " ,
" * 5 -> 1 \n " ,
" * 5, 2, 8, 1, 3 -> 3 "
]
} ,
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
" ## Algorithm \n " ,
" \n " ,
" We ' ll use a recursive algorithm. \n " ,
" \n " ,
" * If the current node is None, return 0 \n " ,
" * Else, return 1 + the maximum height of the left or right children \n " ,
" \n " ,
" Complexity: \n " ,
" * Time: O(n) \n " ,
2015-08-13 18:52:32 +08:00
" * Space: O(h), where h is the height of the tree "
2015-08-02 22:53:35 +08:00
]
} ,
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
" ## Code "
]
} ,
{
" cell_type " : " code " ,
" execution_count " : 1 ,
" metadata " : {
" collapsed " : true
} ,
" outputs " : [ ] ,
" source " : [
" %r un ../bst/bst.py "
]
} ,
{
" cell_type " : " code " ,
" execution_count " : 2 ,
" metadata " : {
" collapsed " : false
} ,
2015-08-06 06:15:03 +08:00
" outputs " : [
{
" name " : " stdout " ,
" output_type " : " stream " ,
" text " : [
2016-06-28 16:33:10 +08:00
" Overwriting height.py \n "
2015-08-06 06:15:03 +08:00
]
}
] ,
2015-08-02 22:53:35 +08:00
" source " : [
2015-08-06 06:15:03 +08:00
" %% writefile height.py \n " ,
2015-08-02 22:53:35 +08:00
" def height(node): \n " ,
" if node is None: \n " ,
" return 0 \n " ,
" return 1 + max(height(node.left), \n " ,
" height(node.right)) "
]
} ,
2015-08-06 06:15:03 +08:00
{
" cell_type " : " code " ,
" execution_count " : 3 ,
" metadata " : {
" collapsed " : true
} ,
" outputs " : [ ] ,
" source " : [
" %r un height.py "
]
} ,
2015-08-02 22:53:35 +08:00
{
" cell_type " : " markdown " ,
" metadata " : { } ,
" source " : [
" ## Unit Test "
]
} ,
{
" cell_type " : " code " ,
2015-08-06 06:15:03 +08:00
" execution_count " : 4 ,
2015-08-02 22:53:35 +08:00
" metadata " : {
" collapsed " : false
} ,
" outputs " : [
{
" name " : " stdout " ,
" output_type " : " stream " ,
" text " : [
" Overwriting test_height.py \n "
]
}
] ,
" source " : [
" %% writefile test_height.py \n " ,
" from nose.tools import assert_equal \n " ,
" \n " ,
" \n " ,
" class TestHeight(object): \n " ,
" \n " ,
" def test_height(self): \n " ,
" root = Node(5) \n " ,
" assert_equal(height(root), 1) \n " ,
" insert(root, 2) \n " ,
" insert(root, 8) \n " ,
" insert(root, 1) \n " ,
" insert(root, 3) \n " ,
" assert_equal(height(root), 3) \n " ,
" \n " ,
" print( ' Success: test_height ' ) \n " ,
" \n " ,
" \n " ,
" def main(): \n " ,
" test = TestHeight() \n " ,
" test.test_height() \n " ,
" \n " ,
" \n " ,
" if __name__ == ' __main__ ' : \n " ,
" main() "
]
} ,
{
" cell_type " : " code " ,
2015-08-06 06:15:03 +08:00
" execution_count " : 5 ,
2015-08-02 22:53:35 +08:00
" metadata " : {
" collapsed " : false
} ,
" outputs " : [
{
" name " : " stdout " ,
" output_type " : " stream " ,
" text " : [
" Success: test_height \n "
]
}
] ,
" source " : [
" %r un -i test_height.py "
]
}
] ,
" metadata " : {
" kernelspec " : {
2016-06-28 16:33:10 +08:00
" display_name " : " Python 3 " ,
2015-08-02 22:53:35 +08:00
" language " : " python " ,
2016-06-28 16:33:10 +08:00
" name " : " python3 "
2015-08-02 22:53:35 +08:00
} ,
" language_info " : {
" codemirror_mode " : {
" name " : " ipython " ,
2016-06-28 16:33:10 +08:00
" version " : 3
2015-08-02 22:53:35 +08:00
} ,
" file_extension " : " .py " ,
" mimetype " : " text/x-python " ,
" name " : " python " ,
" nbconvert_exporter " : " python " ,
2016-06-28 16:33:10 +08:00
" pygments_lexer " : " ipython3 " ,
" version " : " 3.5.0 "
2015-08-02 22:53:35 +08:00
}
} ,
" nbformat " : 4 ,
" nbformat_minor " : 0
}