打印指定层的节点

• 打印指定层的节点

class node():
    def __init__(self,root=None,left=None,right=None):
        self.root=root
        self.left=left
        self.right=right
t=node('a',node('b',node('d'),node('e')),node('c'))
def TransLevel(node,level):
    if node == None:
        return
    else:
        if level == 1 :
            print ("%s " % node.root)
        else:
            
            TransLevel(node.left, level-1)
            TransLevel(node.right, level-1)
    
    
TransLevel(t,3)

• 打印各层节点

class node():
    def __init__(self,root=None,left=None,right=None):
        self.root=root
        self.left=left
        self.right=right
t=node('a',node('b',node('d'),node('e')),node('c'))
def TLevel(node,level,a):
    if node == None:
        return
    else:
            print ("%s " % node.root)
            
            if level in a :
                a[level].append(node.root)
                
            else:
                a[level]=[node.root]
            print(a)     
            TLevel(node.left, level+1,a)
            TLevel(node.right, level+1,a)
a={}
n=1
TLevel(t,n,a)