Algorithms For Dummies

 Understanding the Need to Sort and Search

Building a tree

Python doesn’t come with a built-in tree object. You must either create your own 

implementation or use a tree supplied with a package. A basic tree implementa-

tion requires that you create a class to hold the tree data object. The following code 

shows how you can create a basic tree class. (You can find this code in the 

A4D; 

06; Trees.ipynb

 file on the Dummies site as part of the downloadable code; see 

the Introduction for details.)

class binaryTree:

    def __init__(self, nodeData, left=None, right=None):

        self.nodeData = nodeData

        self.left  = left

        self.right = right

  

    def __str__(self):

        return str(self.nodeData)

All this code does is create a basic tree object that defines the three elements that 

a node must include: data storage, left connection, and right connection. Because 

leaf nodes have no connection, the default value for 

left

 and 

right

 is 

None

. The 

class also includes a method for printing the content of 

nodeData

 so that you can 

see what data the node stores.

Using this simple tree requires that you not try to store anything in 

left

 or 

right

 

other than a reference to another node. Otherwise, the code will fail because there 

FIGURE 6-1: 

A tree in Python 

looks much like 

the physical 

alternative.

---

CHAPTER 6