A document from MCS 275 Spring 2022, instructor David Dumas. You can also get the notebook file.

MCS 275 Spring 2022 Worksheet 8 Solutions

  • Course instructor: David Dumas
  • Solutions prepared by: Johnny Joyce, Jennifer Vaccaro

Topics

The main topics of this worksheet are:

  • Trees
  • Binary search trees

Resources

These things might be helpful while working on the problems. Remember that for worksheets, we don't strictly limit what resources you can consult, so these are only suggestions.

The most useful files from the sample code repository are:

1. Node relationship

Suppose you have two Node objects (or objects of a subclass of Node) as defined in (https://github.com/daviddumas/mcs275spring2022/blob/main/samplecode/trees/trees.py). Let's call them x and y. You might ask how they are related to one another, if it all. In general, exactly one of the following statements will be true:

  1. The nodes are equal
  2. One of the nodes is an ancestor of the other
  3. The nodes lie in the same tree, but neither is an ancestor of the other
  4. The nodes do not lie in the same tree

Write a function node_relationship(x,y) that takes two nodes and returns one of these strings:

  1. "equal"
  2. "ancestor"
  3. "cousin"
  4. "unrelated"

according to which of the cases listed above describes the relationship between x and y.

(Note that the use of "cousin" to describe case (3) doesn't fit the genealogical analogy perfectly, as it would also include the case of siblings.)

Also, make test cases for your function that generate each of the possible answers.

Solution 1:

Uses a helper function parents to find all parents, grandparents, great-grandparents, etc.

In [ ]:
def parents(x)
    '''
    Given node, returns list containing its parent, its parent's parent, etc.
    '''
    L = [x]
    while L[-1].parent != None: # While we can find a parent of the most recently-found parent
        L.append(L[-1].parent) # Put parent into the list
    return L
        
def node_relationship(x,y):
    """
    Returns string repesenting genealogical relationship between nodes x and y
    """
    
    if x == y:
        return "equal"
    
    # Find all parents of x (including its grandparent, great-grandparent, etc.)
    parents_x = parents(x)
    # Find all parents of y
    parents_y = parents(y)
        
    if x in parents_y:
        return "ancestor"
    
    if y in parents_x:
        return "ancestor"
    
    # Last item in each list is root of tree.
    if parents_x[-1] == parents_y[-1]:
        return "cousins"
    
    # If no previous attempts to find relationship worked, then x and y are unrelated.
    return "unrelated"

Solution 2:

Slightly shorter version using keys of nodes. Can fail if given nodes in two separate trees which contain the same key at some point, so avoid reusing code from this version

In [2]:
def node_relationship(x,y):
    """
    Returns string repesenting genealogical relationship between nodes x and y
    """
    
    if x.key == y.key:
        return "equal"
    
    if x.search(y.key) != None: # If y can be found "below" x
        return "ancestor"
    
    if y.search(x.key) != None: # If x can be found "below" y
        return "ancestor"
    
    # Find the root of x's tree
    root_x = x
    while root_x.parent != None: # Keep ascending tree until root is found
        root_x = root_x.parent
        
    if root_x.search(y.key) != None: # If y can be found anywhere in x's tree
        return "cousin"
    
    # If no previous attempts to find relationship worked, then x and y are unrelated.
    return "unrelated"

Get ready to edit trees.py

The rest of the worksheet focuses on adding features to the module trees.py from lecture. To prepare for that, download these files and save them in a directory where you'll work on worksheet 8.

Know how to test your code

It will be hard to test your code if you don't have any search trees to test it with. To that end, I created a module treeutil.py (which you were asked to download above) that will generate random binary search trees on request. It can also give a random node of the tree, or a random pair of nodes.

Open the full documentation of the module in a separate browser tab, and keep it open as you work:

2. Minimum and maximum

Add two new methods, minimum and maximum to the BST class for finding the node with the smallest or largest key in the subtree that has this node as its root.

Solution

In trees.py under the BST class:

In [ ]:
    def minimum(self):
        """
        In the subtree that this node is the root of, find and
        return the the smallest key.
        """

        if self.left != None: # Try to return the min of the left side first (left side is always smaller)
            return self.left.minimum()

        else: # Otherwise, no need to look at right side (because it must be bigger)
            return self.key
In [ ]:
    def maximum(self):
        """
        In the subtree that this node is the root of, find and
        return the the largest key.
        """
        
        if self.right != None: # Try to return the max of the right side first (right side is always bigger)
            return self.right.minimum()

        else: # Otherwise, no need to look at left side (because it must be smaller)
            return self.key

3. Depth

Add a method depth to the BST class that computes the depth of the tree (optionally starting at a certain node rather than the root).

Solution:

In [2]:
# This should go inside class `BST` in `trees.py`
    def depth(self, start_node = None):
        """
        Return the length (in number of edges) of the longest descending path starting
        from the node `start_node`. If `start_node` is not given, starts from root node.
        """
        # If user gave optional argument for starting point
        if start_node != None:
            initial = self.search(start_node)
            return initial.depth()

        # Base case: if there are no child Nodes, cannot go any further down
        if self.left == None and self.right == None:
            return 0

        # Cases where only one child Node is defined
        if self.right == None:
            return self.left.depth() + 1 # Add 1 to account for increased depth
        elif self.left == None:
            return self.right.depth() + 1 # Add 1 to account for increased depth

        # Last case: both left and right child Nodes are defined
        else:
            # Note this is the built-in `max` function, not the one we wrote in Q2
            return max(self.left.depth(), self.right.depth()) + 1    

4. Interval of a node

Suppose you are given a binary search tree T and a node x that has no children. If you change the key of x, the tree may or may not be a binary search tree.

Here's an example to illustrate this. Consider this BST:

       [15]

   [9]      [29]

[5]  [12]

If x is the node with key 12, then changing the key to 13 still gives a binary search tree:

       [15]

   [9]      [29]

[5]  [13]

But if we change the key to 6, the result is not a binary search tree; the right subtree of the node with key 9 contains keys smaller than 9:

❌THIS IS NOT A BST❌
       [15]

   [9]      [29]

[5]  [6]

And if we change the key to 18, the result is not a binary search tree. The left subtree of the root node contains a node with key larger than 15:

❌THIS IS NOT A BST❌
       [15]

   [9]      [29]

[5]  [18]

Now, if you look more closely at this example, you can convince yourself that the key 12 can be changed to any number in the closed interval $[9,15]$ while keeping the BST condition, but that anything outside of this range will result in a non-BST. This is called the interval of the node.

Add a method to BST that can be called whenever the node has no children, and which returns a tuple (kmin,kmax) where kmin and kmax are the smallest and largest values (respectively) that the key of that node could be changed to while keeping the BST condition. In some cases, there may be no lower or upper limit, in which case the value None should be returned for kmin, kmax, or both. If called on a node that has children, it should raise an exception.

Use this method definition:

In [ ]:
# This should go inside class `BST` in `trees.py`
    def interval(self):
        """
        If this node has no children, return the minimum and maximum key values that
        could be given to this node without violating the BST condition.  The value None
        is used to indicate that the range of allowable values has no upper or lower
        bound.
        """

For example, in this tree:

       [15]

   [9]      [29]

[5]  [12]

calling .interval() on its nodes will result in the following return values:

[15] : (raises exception because the node has children)
 [9] : (raises exception because the node has children)
 [5] : None,9
[12] : 9,15
[29] : 15,None

Finally, it may be helpful in this problem to recall that functions in Python can return multiple values, and if they do, the returned values are assembled into a tuple. That means the two statements below do the same thing:

In [ ]:
    return 1,2,3
    return (1,2,3)

Solution:

In [ ]:
    def interval(self):
        """
        If this node has no children, return the minimum and maximum key values that
        could be given to this node without violating the BST condition.  The value None
        is used to indicate that the range of allowable values has no upper or lower
        bound.
        """
        if self.left != None or self.right != None:
            raise ValueError("Method `interval` may only be used on nodes with no children")
        
        # Left and right sides of interval to be returned
        interval_left = None
        interval_right = None

        if self.parent == None: # If we are at the root of the tree, then interval is unlimited
            return (interval_left, interval_right)

        # If current node is to the parent's left
        if self.parent.left.key == self.key:
            interval_right = self.parent.key # Upper bound is given by parent's key
            if self.parent.parent != None:
                grandparent = self.parent.parent
                # Check if parent is to the right of its own parent
                if grandparent.right.key == self.parent.key:
                    # If so, then parent's parent's key is lower bound
                    interval_left = grandparent.key
        
        # Else, current node is to the parent's right
        else:
            interval_left = self.parent.key # Lower bound is given by parent's key
            if self.parent.parent != None:
                grandparent = self.parent.parent
                # Check if parent is to the left of its own parent
                if grandparent.left.key == self.parent.key:
                    # If so, then parent's parent's key is upper bound
                    interval_right = grandparent.key
                    
        return (interval_left, interval_right)

Code to test the tree using the given example:

In [1]:
import treevis
import treeutil

T = treeutil.BST(15)
T.insert(9)
T.insert(29)
T.insert(5)
T.insert(12)
treevis.treeprint(T)

for key in [5, 12, 29]: # Check interval of each of: 5, 12, and 29
    node = T.search(key)
    print("Interval of {}: {}".format(node, node.interval()))
      [15]

  [9]     [29]

[5] [12]  .   .

Interval of [5]: (None, 9)
Interval of [12]: (9, 15)
Interval of [29]: (15, None)

Challenge

This is an extra activity to work on if you finish all the exercises above. This won't be included in the worksheet solutions.

Given a node x in a BST, you might want to know its successor, meaning the node whose key is larger than x but as small as possible given that condition. (We'll assume in this problem that all nodes in the tree have distinct keys).

It turns out that there is an efficient description of the successor as follows:

  • If x has a right child, then the successor is the smallest key in the right subtree of x
  • If x does not have a right child, but has a successor, then the successor is an ancestor of x. If you visit these ancestors by traveling upward from x, then the first node you reach that is the left child of its parent is the successor.
  • If x has no right child, and if moving up from x to the root only ever visits nodes that are right children of their parents, then x is the maximum of the tree, so it has no successor.

Write a method successor of class BST that finds and returns the successor of a node, or returns None if there is no successor.

Revision history

  • 2022-02-26 - Initial publication