Python实现简单二叉树

class BinaryTree:
    def __init__(self,rootObj):
        self.root = rootObj
        self.leftChild = None
        self.rightChild = None

    def insertLeft(self,newNode):
        if self.leftChild == None:
            self.leftChild = BinaryTree(newNode)
        else:
            print('The leftChild is not None.You can not insert')
    def insertRight(self,newNode):
        if self.rightChild == None:
            self.rightChild = BinaryTree(newNode)
        else:
            print('The rightChild is not None.You can not insert')

构建了一个简单的二叉树类，它的初始化函数，将传入的rootObj赋值给self.root，作为根节点，leftChild和rightChild都默认为None。
函数insertLeft为向二叉树的左子树赋值，若leftChild为空，则先构造一个BinaryTree(newNode)，即实例化一个新的二叉树，然后将这棵二叉树赋值给原来的二叉树的leftChild。此处递归调用了BinaryTree这个类。
若不为空则输出：The rightChild is not None.You can not insert

执行下述语句

r = BinaryTree('a')
print('root:',r.root,';','leftChild:',r.leftChild,';','rightChild:',r.rightChild)

输出

root: a ; leftChild: None ; rightChild: None

即我们构造了一颗二叉树，根节点为a，左右子树均为None

然后执行下述语句

r.insertLeft('b')
print('root:',r.root,';','leftChild:',r.leftChild,';','rightChild:',r.rightChild)
print('root:',r.root,';','leftChild.root:',r.leftChild.root,';','rightChild:',r.rightChild)

输出

root: a ; leftChild: <__main__.BinaryTree object at 0x000002431E4A0DA0> ; rightChild: None
root: a ; leftChild.root: b ; rightChild: None

我们向r插入了一个左节点，查看输出的第一句话，可以看到左节点其实也是一个BinaryTree，这是因为插入时，递归生成的。
第二句输出，可以查看左节点的值

最后执行

r.insertLeft('c')

The leftChild is not None.You can not insert

可以看到，我们无法再向左节点插入了，因为该节点已经有值了

Python实现简单二叉树

猜你喜欢