Python简单定义与使用二叉树示例

时间：2021-12-05 10:18:53|栏目：Python代码|点击：次

本文实例讲述了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)

输出