Saved tree height challenge solution so it can be reused as a component to other related challenges.

This commit is contained in:
Donne Martin 2015-08-05 18:15:03 -04:00
parent b109b6f8cc
commit d098e67ffd
2 changed files with 28 additions and 3 deletions

View File

@ -0,0 +1,5 @@
def height(node):
if node is None:
return 0
return 1 + max(height(node.left),
height(node.right))

View File

@ -89,8 +89,17 @@
"metadata": { "metadata": {
"collapsed": false "collapsed": false
}, },
"outputs": [], "outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Writing height.py\n"
]
}
],
"source": [ "source": [
"%%writefile height.py\n",
"def height(node):\n", "def height(node):\n",
" if node is None:\n", " if node is None:\n",
" return 0\n", " return 0\n",
@ -98,6 +107,17 @@
" height(node.right))" " height(node.right))"
] ]
}, },
{
"cell_type": "code",
"execution_count": 3,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"%run height.py"
]
},
{ {
"cell_type": "markdown", "cell_type": "markdown",
"metadata": {}, "metadata": {},
@ -107,7 +127,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 3, "execution_count": 4,
"metadata": { "metadata": {
"collapsed": false "collapsed": false
}, },
@ -150,7 +170,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 4, "execution_count": 5,
"metadata": { "metadata": {
"collapsed": false "collapsed": false
}, },