按层序生成二叉树（完全二叉树） python

一、思路

按层序生成的二叉树一定是完全二叉树。
完全二叉树的特点：叶子结点只能出现在最下层和次下层，且最下层的叶子结点集中在树的左部（满二叉树肯定是完全二叉树，而完全二叉树不一定是满二叉树）。
那么，如果输入 [1,2,3,4,5,6,7]，生成的树就应该是：

1、节点结构
一个节点有一个值域，两个指针域，结构就是：

class Node:
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None

2、树的结构
data是初始化时的参数，类型是数组

class TwoTree:
    def __init__(self, data:list):
	    self.root = None # 	根节点

3、添加节点
1. 当根节点是None时，直接实例化Node赋值给root
2. 当根节点非空，左指针域是空时，实例化Node赋值给左指针
3. 当根节点非空左指针非空，右指针是空时，实例化Node赋值给右指针
当上述三种情况都不满足时，说明此时有三个节点，形状为

第四个节点应该要添加到值为2的节点的左指针上
第五个节点应该要添加到值为2的节点的右指针上
第六个节点应该要添加到值为3的节点的左指针上
…以此类推

最终发现，从左往右依次添加，先添加的节点优先为其添加子节点，（先判断左指针是否为空，如果为空则添加，在判断右指针，所以右节点一定在左节点之后添加）。这性质和队列一致（先进先出），所以可以使用队列来实现。

每当添加一个节点，都入队。由于每个节点有两个指针域（左右节点），所以同一个节点要入队两次，第一次添加左节点，第二次添加右节点。

    # 添加一个节点
    def addNode(self, val):
        node = Node(val)  # 实例化一个节点
        if not self.root: # 如果根节点是空，上述的第一种情况
            self.root = node
            self._q.put(self.root)
            self._q.put(self.root)
            return
        n = self._q.get()	# 获取父节点，为其添加左右节点
        if n.left is None:	# 第二种情况
            n.left = node
            self._q.put(node)
            self._q.put(node)
            return
        elif n.right is None:	#	第三种情况
            n.right = node
            self._q.put(node)
            self._q.put(node)

之后再构造函数只需要依次调用addNode函数即可构造一棵完全二叉树。

    def __init__(self, data:list):
        self._q = queue.Queue()	# 	设置为私有变量
        self.root = None		#	根节点
        for i in data:			#	构造树
            self.addNode(i)

4、打印树的值
实现先序遍历（根左右），如果是含有7个节点的树，那么打印的值就应该是：1 2 4 5 3 6 7

    #   先序遍历
    def Foutput(self, node):
    #	每次都将node作为根节点，如果node非None，所以print在递归的前面
        if node is None:
            return
        print(node.val, end="\t")
        self.Foutput(node.left)
        self.Foutput(node.right)

代码的运行结果:

二、代码

整体代码为

import queue

class Node:
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None

    def __str__(self):
        return f'<val={self.val} left={self.left} right={self.right}> '

class TwoTree:
    def __init__(self, data:list):
        self._q = queue.Queue()
        self.root = None
        for i in data:
            self.addNode(i)

    # 添加一个节点
    def addNode(self, val):
        node = Node(val)
        if not self.root:
            self.root = node
            self._q.put(self.root)
            self._q.put(self.root)
            return
        n = self._q.get()
        if n.left is None:
            n.left = node
            self._q.put(node)
            self._q.put(node)
            return
        elif n.right is None:
            n.right = node
            self._q.put(node)
            self._q.put(node)

    #   先序遍历
    def Foutput(self, node):
        if node is None:
            return
        print(node.val, end="\t")
        self.Foutput(node.left)
        self.Foutput(node.right)

if __name__ == '__main__':
    tree = TwoTree([1,2,3,4,5,6,7])
    tree.Foutput(tree.root)
    print("\n", tree.root)
    tree.addNode(8)	##	添加一个节点,该节点应添加在值为4的左节点
    tree.Foutput(tree.root)
    print("\n", tree.root)