团体程序设计天梯赛 L3-010 是否完全二叉搜索树 （30）（Python3）

问题描述： 

L3-010 是否完全二叉搜索树 （30 分)

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

输入格式：

输入第一行给出一个不超过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

AC代码： 

import queue
input()
l=list(map(int,input().split()))
bt=[]
for i in l:
	r=0
	if r<len(bt):
		while(True):
			if i>bt[r][0]:
				if bt[r][1]<0:
					bt[r][1]=len(bt)
					break
				else:
					r=bt[r][1]
			else:
				if bt[r][2]<0:
					bt[r][2]=len(bt)
					break
				else:
					r=bt[r][2]
	bt.append([i,-1,-1])

q=queue.Queue()
q.put(0)
flag=0
print(bt[0][0],end='')
while not q.empty():
	t=q.get()
	if bt[t][1]>=0:
		q.put(bt[t][1])
		print('',bt[bt[t][1]][0],end='')
		if flag==1:
			flag=2
	elif flag==0:
		flag=1
	if bt[t][2]>=0:
		q.put(bt[t][2])
		print('',bt[bt[t][2]][0],end='')
		if flag==1:
			flag=2
	elif flag==0:
		flag=1
print('')
if flag==2:
	print('NO')
else:
	print('YES')