Dis05
Trees
Q1: Tree Abstraction Barrier
Consider a tree t
constructed by calling tree(1, [tree(2), tree(4)])
.
- What does the expression evaluate to?
- Does the expression violate any abstraction barriers? If so, write an equivalent expression that does not violate abstraction barriers.
1.label(t)
1
2.t[0]
1;label(t)
3.label(branches(t)[0])
2
4.is_leaf(t[1:][1])
True;is_leaf(branches(t)[1])
5.[label(b) for b in branches(t)]
[2,4]
6.branches(tree(5, [t, tree(3)]))[0][0]
1;label(branches(tree(5, [t, tree(3)]))[0])
Q2: Height
Write a function that returns the height of a tree. Recall that the height of a tree is the length of the longest path from the root to a leaf.
def height(t):
"""Return the height of a tree.
>>> t = tree(3, [tree(5, [tree(1)]), tree(2)])
>>> height(t)
2
>>> t = tree(3, [tree(1), tree(2, [tree(5, [tree(6)]), tree(1)])])
>>> height(t)
3
"""
if is_leaf(t):
return 0
return max([1 + height(_) for _ in branches(t)])
Q3: Maximum Path Sum
Write a function that takes in a tree and returns the maximum sum of the values along any path in the tree. Recall that a path is from the tree's root to any leaf.
def max_path_sum(t):
"""Return the maximum path sum of the tree.
>>> t = tree(1, [tree(5, [tree(1), tree(3)]), tree(10)])
>>> max_path_sum(t)
11
"""
if is_leaf(t):
return label(t)
return max([label(t)+max_path_sum(_) for _ in branches(t)])
Q4: Find Path
Write a function that takes in a tree and a value x
and returns a list containing the nodes along the path required to get from the root of the tree to a node containing x
.
If x
is not present in the tree, return None
. Assume that the entries of the tree are unique.
def find_path(t, x):
"""
>>> t = tree(2, [tree(7, [tree(3), tree(6, [tree(5), tree(11)])] ), tree(15)])
>>> find_path(t, 5)
[2, 7, 6, 5]
>>> find_path(t, 10) # returns None
"""
if label(t) == x or is_leaf(t):
return [label(t)]
for _ in branches(t):
path = [label(t)] + find_path(_, x)
if path[-1] == x:
return path
Sequences
Q5: Map, Filter, Reduce
Many languages provide map
, filter
, reduce
functions for sequences. Python also provides these functions (and we'll formally introduce them later on in the course), but to help you better understand how they work, you'll be implementing these functions in the following problems.
def my_map(fn, seq):
"""Applies fn onto each element in seq and returns a list.
>>> my_map(lambda x: x*x, [1, 2, 3])
[1, 4, 9]
"""
return [fn(_) for _ in seq]
def my_filter(pred, seq):
"""Keeps elements in seq only if they satisfy pred.
>>> my_filter(lambda x: x % 2 == 0, [1, 2, 3, 4]) # new list has only even-valued elements
[2, 4]
"""
return [_ for _ in seq if pred(_)]
def my_reduce(combiner, seq):
"""Combines elements in seq using combiner.
seq will have at least one element.
>>> my_reduce(lambda x, y: x + y, [1, 2, 3, 4]) # 1 + 2 + 3 + 4
10
>>> my_reduce(lambda x, y: x * y, [1, 2, 3, 4]) # 1 * 2 * 3 * 4
24
>>> my_reduce(lambda x, y: x * y, [4])
4
>>> my_reduce(lambda x, y: x + 2 * y, [1, 2, 3]) # (1 + 2 * 2) + 2 * 3
11
"""
if len(seq) == 1:
return seq[0]
return combiner(seq[0], my_reduce(combiner, seq[1:]))
Q6: Count palindromes
Write a function that counts the number of palindromes (any word that reads the same forwards as it does when read backwards) in a list of words using only lambda
, string operations, conditional expressions, and the functions we defined above (my_filter
, my_map
, my_reduce
). Specifically, do not use recursion or any kind of loop.
def count_palindromes(L):
"""The number of palindromic words in the sequence of strings
L (ignoring case).
>>> count_palindromes(("Acme", "Madam", "Pivot", "Pip"))
2
"""
return my_reduce(lambda x, y: x + y,
my_map(lambda s:1,
my_filter(lambda s: s.lower() == s.lower()[::-1] ,L)))
Additional Practice
Q7: Perfectly Balanced
Part A: Implement sum_tree
, which returns the sum of all the labels in tree t
.
Part B: Implement balanced
, which returns whether every branch of t
has the same total sum and that the branches themselves are also balanced.
Challenge: Solve both of these parts with just 1 line of code each.
def sum_tree(t):
"""
Add all elements in a tree.
>>> t = tree(4, [tree(2, [tree(3)]), tree(6)])
>>> sum_tree(t)
15
"""
return label(t) + sum([sum_tree(_) for _ in branches(t)])
def balanced(t):
"""
Checks if each branch has same sum of all elements and
if each branch is balanced.
>>> t = tree(1, [tree(3), tree(1, [tree(2)]), tree(1, [tree(1), tree(1)])])
>>> balanced(t)
True
>>> t = tree(1, [t, tree(1)])
>>> balanced(t)
False
>>> t = tree(1, [tree(4), tree(1, [tree(2), tree(1)]), tree(1, [tree(3)])])
>>> balanced(t)
False
"""
branc = branches(t)
sum = [sum_tree(_) for _ in branc]
for _ in range(len(sum)):
if sum[0] != sum[_] or not balanced(branc[_]):
return False
return True
Q8: Hailstone Tree
We can represent the hailstone sequence as a tree in the figure below, showing the route different numbers take to reach 1. Remember that a hailstone sequence starts with a number n
, continuing to n/2
if n
is even or 3n+1
if n
is odd, ending with 1. Write a function hailstone_tree(n, h)
which generates a tree of height h
, containing hailstone numbers that will reach n
.
def hailstone_tree(n, h):
"""Generates a tree of hailstone numbers that will reach N, with height H.
>>> print_tree(hailstone_tree(1, 0))
1
>>> print_tree(hailstone_tree(1, 4))
1
2
4
8
16
>>> print_tree(hailstone_tree(8, 3))
8
16
32
64
5
10
"""
if h == 0:
return tree(n)
branches = [hailstone_tree(n*2, h-1)]
if (n-1) % 3 == 0 and n != 4 and n != 1:
branches += [hailstone_tree((n-1) // 3, h-1)]
return tree(n, branches)
def print_tree(t):
def helper(i, t):
print(" " * i + str(label(t)))
for b in branches(t):
helper(i + 1, b)
helper(0, t)