Lab 8: Mutable Trees
Due by 11:59pm on Wednesday, March 20.
Starter Files
Download lab08.zip. Inside the archive, you will find starter files for the questions in this lab, along with a copy of the Ok autograder.
Topics
Consult this section if you need a refresher on the material for this lab. It's okay to skip directly to the questions and refer back here should you get stuck.
A Tree
instance has two instance attributes:
label
is the value stored at the root of the tree.branches
is a list ofTree
instances that hold the labels in the rest of the tree.
The Tree
class (with its __repr__
and __str__
methods omitted) is defined as:
class Tree: """ >>> t = Tree(3, [Tree(2, [Tree(5)]), Tree(4)]) >>> t.label 3 >>> t.branches[0].label 2 >>> t.branches[1].is_leaf() True """ def __init__(self, label, branches=[]): for b in branches: assert isinstance(b, Tree) self.label = label self.branches = list(branches) def is_leaf(self): return not self.branches
To construct a Tree
instance from a label x
(any value) and a list of branches bs
(a list of Tree
instances) and give it the name t
, write t = Tree(x, bs)
.
For a tree t
:
- Its root label can be any value, and
t.label
evaluates to it. - Its branches are always
Tree
instances, andt.branches
evaluates to the list of its branches. t.is_leaf()
returnsTrue
ift.branches
is empty andFalse
otherwise.- To construct a leaf with label
x
, writeTree(x)
.
Displaying a tree t
:
-
repr(t)
returns a Python expression that evaluates to an equivalent tree. -
str(t)
returns one line for each label indented once more than its parent with children below their parents.t = Tree(3, [Tree(1, [Tree(4), Tree(1)]), Tree(5, [Tree(9)])])>>> t # displays the contents of repr(t)Tree(3, [Tree(1, [Tree(4), Tree(1)]), Tree(5, [Tree(9)])])>>> print(t) # displays the contents of str(t)3 1 4 1 5 9
Changing (also known as mutating) a tree t
:
-
t.label = y
changes the root label oft
toy
(any value). -
t.branches = ns
changes the branches oft
tons
(a list ofTree
instances). -
Mutation of
t.branches
will changet
. For example,t.branches.append(Tree(y))
will add a leaf labeledy
as the right-most branch. -
Mutation of any branch in
t
will changet
. For example,t.branches[0].label = y
will change the root label of the left-most branch toy
.t.label = 3.0>>> t.branches[1].label = 5.0>>> t.branches.append(Tree(2, [Tree(6)]))>>> print(t)3.0 1 4 1 5.0 9 2 6
Here is a summary of the differences between the tree data abstraction implemented as a functional abstraction vs. implemented as a class:
-
Tree constructor and selector functions
Tree class
Constructing a tree
To construct a tree given a label
and a list of branches
, we call tree(label, branches)
To construct a tree object given a label
and a list of branches
, we call Tree(label, branches)
(which calls the Tree.__init__
method).
Label and branches
To get the label or branches of a tree t
, we call label(t)
or branches(t)
respectively
To get the label or branches of a tree t
, we access the instance attributes t.label
or t.branches
respectively.
Mutability
The functional tree data abstraction is immutable (without violating its abstraction barrier) because we cannot assign values to call expressions
The label
and branches
attributes of a Tree
instance can be reassigned, mutating the tree.
Checking if a tree is a leaf
To check whether a tree t
is a leaf, we call the function is_leaf(t)
To check whether a tree t
is a leaf, we call the method t.is_leaf()
. This method can only be called on Tree
objects.
Required Questions
Getting Started Videos
These videos may provide some helpful direction for tackling the coding problems on this assignment.
To see these videos, you should be logged into your berkeley.edu email.
Mutable Trees
Q1: WWPD: Trees
Read over the Tree
class in lab08.py
. Make sure you understand the doctests.
Use Ok to test your knowledge with the following "What Would Python Display?" questions:
python3 ok -q trees-wwpd -u
Enter
Function
if you believe the answer is<function ...>
,Error
if it errors, andNothing
if nothing is displayed. Recall thatTree
instances will be displayed the same way they are constructed.
>>> t = Tree(1, Tree(2))______Error>>> t = Tree(1, [Tree(2)])>>> t.label______1>>> t.branches[0]______Tree(2)>>> t.branches[0].label______2>>> t.label = t.branches[0].label>>> t______Tree(2, [Tree(2)])>>> t.branches.append(Tree(4, [Tree(8)]))>>> len(t.branches)______2>>> t.branches[0]______Tree(2)>>> t.branches[1]______Tree(4, [Tree(8)])
Q2: Cumulative Mul
Write a function cumulative_mul
that mutates the Tree t
so that each node's label is replaced by the product of its label and the labels of all its descendents.
Hint: Be careful of the order in which you mutate the current node's label and process its subtrees; which one should come first?
def cumulative_mul(t): """Mutates t so that each node's label becomes the product of its own label and all labels in the corresponding subtree rooted at t. >>> t = Tree(1, [Tree(3, [Tree(5)]), Tree(7)]) >>> cumulative_mul(t) >>> t Tree(105, [Tree(15, [Tree(5)]), Tree(7)]) >>> otherTree = Tree(2, [Tree(1, [Tree(3), Tree(4), Tree(5)]), Tree(6, [Tree(7)])]) >>> cumulative_mul(otherTree) >>> otherTree Tree(5040, [Tree(60, [Tree(3), Tree(4), Tree(5)]), Tree(42, [Tree(7)])]) """ "*** YOUR CODE HERE ***"
Use Ok to test your code:
python3 ok -q cumulative_mul
Q3: Prune Small
Removing some nodes from a tree is called pruning the tree.
Complete the function prune_small
that takes in a Tree
t
and a number n
. For each node with more than n
branches, keep only the n
branches with the smallest labels and remove (prune) the rest.
Hint: The
max
function takes in aniterable
as well as an optionalkey
argument (which takes in a one-argument function). For example,max([-7, 2, -1], key=abs)
would return-7
sinceabs(-7)
is greater thanabs(2)
andabs(-1)
.
def prune_small(t, n): """Prune the tree mutatively, keeping only the n branches of each node with the smallest labels. >>> t1 = Tree(6) >>> prune_small(t1, 2) >>> t1 Tree(6) >>> t2 = Tree(6, [Tree(3), Tree(4)]) >>> prune_small(t2, 1) >>> t2 Tree(6, [Tree(3)]) >>> t3 = Tree(6, [Tree(1), Tree(3, [Tree(1), Tree(2), Tree(3)]), Tree(5, [Tree(3), Tree(4)])]) >>> prune_small(t3, 2) >>> t3 Tree(6, [Tree(1), Tree(3, [Tree(1), Tree(2)])]) """ while ___________________________: largest = max(_______________, key=____________________) _________________________ for __ in _____________: ___________________
Use Ok to test your code:
python3 ok -q prune_small
Check Your Score Locally
You can locally check your score on each question of this assignment by running
python3 ok --score
This does NOT submit the assignment! When you are satisfied with your score, submit the assignment to Gradescope to receive credit for it.
Submit
Submit this assignment by uploading any files you've edited to the appropriate Gradescope assignment. Lab 00 has detailed instructions.
In addition, all students who are not in the mega lab must complete this attendance form. Submit this form each week, whether you attend lab or missed it for a good reason. The attendance form is not required for mega section students.
Optional Questions
Q4: Delete
Implement delete
, which takes a Tree t
and removes all non-root nodes labeled x
. The parent of each remaining node is its nearest ancestor that was not removed. The root node is never removed, even if its label is x
.
def delete(t, x): """Remove all nodes labeled x below the root within Tree t. When a non-leaf node is deleted, the deleted node's children become children of its parent. The root node will never be removed. >>> t = Tree(3, [Tree(2, [Tree(2), Tree(2)]), Tree(2), Tree(2, [Tree(2, [Tree(2), Tree(2)])])]) >>> delete(t, 2) >>> t Tree(3) >>> t = Tree(1, [Tree(2, [Tree(4, [Tree(2)]), Tree(5)]), Tree(3, [Tree(6), Tree(2)]), Tree(4)]) >>> delete(t, 2) >>> t Tree(1, [Tree(4), Tree(5), Tree(3, [Tree(6)]), Tree(4)]) >>> t = Tree(1, [Tree(2, [Tree(4), Tree(5)]), Tree(3, [Tree(6), Tree(2)]), Tree(2, [Tree(6), Tree(2), Tree(7), Tree(8)]), Tree(4)]) >>> delete(t, 2) >>> t Tree(1, [Tree(4), Tree(5), Tree(3, [Tree(6)]), Tree(6), Tree(7), Tree(8), Tree(4)]) """ new_branches = [] for _________ in ________________: _______________________ if b.label == x: __________________________________ else: __________________________________ t.branches = ___________________
Use Ok to test your code:
python3 ok -q delete