給定一個(gè)元素序列(如列表)，遞歸的創(chuàng)建一顆完全二叉樹

完整代碼如下

#! /usr/bin/env python3


class TreeNode:
    """ Node of complete tree"""
    def __init__(self, data=0):
        self.data = data
        self.left = None
        self.right = None


def build(i, array):
    """ Create a compete tree based a given array"""
    if i < len(array):
        tree = TreeNode() 
        tree.data = array[i]
        tree.left = build(2*i + 1, array)
        tree.right = build(2*i + 2, array)
        return tree
    else:
        return None


def in_order(tree):
    """ traversal complete tree in order"""
    if tree != None:
        in_order(tree.left)
        print(tree.data, end=' ')
        in_order(tree.right)


def pre_order(tree):
    """ traversal complete tree pre order"""
    if tree != None:
        print(tree.data, end=' ')
        pre_order(tree.left)
        pre_order(tree.right)


def main():
    t = build(0, [4, 9, 1, 8, 3, 2, 10])
    in_order(t)
    print()
    pre_order(t)
    print()


if __name__ == '__main__':
    main()

程序的運(yùn)行結(jié)果如下