python----二叉树实现及相关操作

一、二叉树子节点个数

#初始化叶子节点和根结点
class treeNode():
    def __init__(self,data=-1,left=None,right=None):
        self.data=data
        self.left=left
        self.right=right
    def __str__(self):
        return str(self.data)
#递归计算包括根节点在内的所有节点个数
def sd_Node(root):
    if root is None:
        return 0
    else:
        left=sd_Node(root.left)
        right=sd_Node(root.right)
        return left+right+1
#测试
if __name__ == '__main__':
    root = treeNode(1,
                    treeNode(2, treeNode(9), treeNode(4)),
                    treeNode(5))
    sd=sd_Node(root)
    print(sd)

二、二叉树深度

 #初始化叶子节点和根结点
 class treeNode():
     def __init__(self,data,left=None,right=None):
         self.data=data
         self.left=left
         self.right=right
     def __str__(self):
         return str(self.data)
 #遍历二叉树的左右分支，比较大小，返回最大的值，即为深度
 def sd_Node(root):
     #判断是否为None
     if root is None:
         return 0
     else:
         #左分支深度
         left=sd_Node(root.left)+1
         #右分支深度
         right=sd_Node(root.right)+1
         #比较，返回
         if left > right:
             return left
         else:
             return right
 if __name__ == '__main__':
     root = treeNode(1,treeNode(1),treeNode(1))
     sd=sd_Node(root)
     print(sd)

三、二叉树反转

 #初始化叶子节点和根结点
 class treeNode():
     def __init__(self,data,left=None,right=None):
         self.data=data
         self.left=left
         self.right=right
     def __str__(self):
         return str(self.data)
 #递归反转左右节点
 def fz_tree(root):
     if root is None:
         return
     else:
         root.left,root.right=fz_tree(root.right),fz_tree(root.left)
         return  root
 #前序遍历输出
 def s_print(root):
     if root:
         print(root.data)
         s_print(root.left)
         s_print(root.right)
 #测试
 if __name__ == '__main__':
     root = treeNode(1,
                     treeNode(2,treeNode(3),treeNode(4)),
                     treeNode(5,treeNode(6)))
     rot=fz_tree(root)
     s_print(rot)