python代码打印二叉树某一特定层的节点

Posted 2021-06-28 Data+Science+Insight

python代码打印二叉树某一特定层的节点

 

#Python实现打印二叉树某一层的所有节点

# 定义二叉树节点类
class Node(object):
    def __init__(self,data=0,left=0,right=0):
        self.data = data
        self.left = left
        self.right = right

# 遍历某一层所有节点，并打印
def TransLevel(root,level):
    if root == None:
        return
    else:
        if level == 1 and  isinstance(root, TreeNode):
            print("%s " % root.data),
        else:
            if isinstance(root, TreeNode):
                TransLevel(root.left, level-1)
                TransLevel(root.right, level-1)

 

#

# 建立的二叉树
# ------------------------
#          root
#       7        8
#     6
#   2   5
# 1    3 4
#
# ------------