7-1 是否完全二叉搜索树 (python)

将一系列给定数字顺序插入一个初始为空的二叉搜索树（定义为左子树键值大，右子树键值小），你需要判断最后的树是否一棵完全二叉树，并且给出其层序遍历的结果。

输入格式：

输入第一行给出一个不超过20的正整数N；第二行给出N个互不相同的正整数，其间以空格分隔。

输出格式：

将输入的N个正整数顺序插入一个初始为空的二叉搜索树。在第一行中输出结果树的层序遍历结果，数字间以1个空格分隔，行的首尾不得有多余空格。第二行输出YES，如果该树是完全二叉树；否则输出NO。

输入样例1：

9
38 45 42 24 58 30 67 12 51

输出样例1：

38 45 24 58 42 30 12 67 51
YES

输入样例2：

8
38 24 12 45 58 67 42 51

输出样例2：

38 45 24 58 42 12 67 51
NO

代码实现：

 

class treeNode():  #节点类，有三个属性，值，左节点，右节点
    def __init__(self,val,lchild=None,rchild=None):
        self.val = val
        self.lchild = lchild 
        self.rchild = rchild 
        
class BinaryStree():       #二叉搜索树
    def __init__(self):    #初始化函数
        self.root = None   #初始化为一颗空二叉树
    def insert(self,root,val):    #用递归插入来创建二叉树
        if root == None:          #如果根引用指向为空
            root = treeNode(val)  #将被插入元素初始化为节点，赋给根引用指向的节点
        elif val>root.val:        #如果被插入元素比当前根节点的值小
            root.lchild = self.insert(root.lchild,val)  #往它的左子树中插入，调用自身函数递归实现
        elif val<root.val:   #同上
            root.rchild = self.insert(root.rchild,val)
        return root   #要修改根，所以要返回
    
    
    
    def level_travel(self,root):   #层次遍历
        if root == None:        #如果为空二叉树
            return              #直接返回
        q = []            #创建一个空列表
        val_list = []     #用来加入每个cur值
        q.append(root)    #将根节点入队
        while q:     
            cur = q.pop(0)  #将队列首个元素弹出，赋给cur
            val_list.append(cur.val)  #将cur加到val_list列表，方便后续输出
            if cur.lchild != None:    #如果弹出根节点的左孩子不为空
                q.append(cur.lchild)  #就把左孩子加到队列q中
            if cur.rchild != None:    #同上
                q.append(cur.rchild)
#使得每个根节点的左孩子入队时总在右孩子前，一直循环遍历，直到队列q为空
#例：[7,5,8,4,6,9,10]
#将根节点7入队，再弹出7并返回(q中不再有7)，看它的左右节点是否为空，不为空再将左节点5先入队，右节点8后入队，即[5,8]
#q不为空就一直遍历，继续上面，后将5弹出并返回，左右不空，继续入队，即[8,4,6]
#8弹出并返回，左右不空，9，10，入队，即[4,6,9,10]
#4，6，9，10都没有孩子节点，直接逐个弹出并返回，直至q为空
        val_list = [str(x) for x in val_list]   #将列表值转换成字符串
        print (" ".join(val_list))     #用join函数输出字符以空格间隔
        

#是不是完全二叉树
#左无，右有 ==> 返回False
#左无，右无 ==> 激活判断:之后所有节点都是叶节点
#左有，右无 ==> 激活判断:之后所有节点都是叶节点  ==>由此可知只要右无就激活判断
#左有，右有 ==> 不用处理

def isCBTree(root):
    if root == None:
        return False
    que = []
    que.append(root)
    flag = False  #激活判断的标志
    while que:
        cur = que.pop(0)
        if cur.lchild != None:
            que.append(cur.lchild)
        if cur.rchild != None:
            que.append(cur.rchild)
            
        if cur.lchild == None and cur.rchild:  #左无，右有
            return False
        
        if flag:
            if cur.lchild or cur.rchild:  #判断之后节点是不是叶节点
                return False
        if not cur.rchild:  #右无，激活判断
            flag = True
    return True

if __name__ == "__main__":
    bst0 = BinaryStree()  #实例化一颗二叉搜索树
    n = eval(input())
    ln = list(map(eval,input().split()))
    
    for elem in ln:
        bst0.root = bst0.insert(bst0.root,elem)  #将输入元素插入，形成二叉搜索树
    bst0.level_travel(bst0.root)  #层次遍历
    if isCBTree(bst0.root):   #如果是完全二叉树
        print("YES")   #输出yes
    else:              #不是
        print("NO")    #输出no