完全二叉树的创建与遍历

完全二叉树:如果一个树的高度是k(k>1),树的前k-1层是满二叉树,并且第k层的叶子结点都是靠左排列。

在本例中树结点的val值为1至n-1,可以根据需求自行更改。

def createTree(x,n):#创建完全二叉树
    if x<n:
        node=TreeNode(x)#x传参为1,表示根节点为1
        node.left=createTree(2*x,n)
        node.right=createTree(2*x+1,n)
    else:   
        node=None
    return node

当我们调用createTree时设置x为1,n为10,创建的树结构如下所示。

接下来,我们会分别使用先序、中序、后序进行遍历。其共同点是都依靠遍历来实现,都是采用先遍历左子树,再遍历右子树。区别在于结点值输出的位置不同。

先序遍历:

def printTree(node):#先序遍历
    if node!=None:
        print(node.val,end=" ")
        printTree(node.left)
        printTree(node.right)

中序遍历:

def printMidTree(node):#中序遍历
    if node!=None:
        printMidTree(node.left)
        print(node.val, end=" ")
        printMidTree(node.right)

后序遍历:

def printLastTree(node):#后序遍历
    if node!=None:
        printLastTree(node.left)
        printLastTree(node.right)
        print(node.val,end=" ")

猜你喜欢

转载自blog.csdn.net/noingw96/article/details/84990054
今日推荐