原题来源PTA
给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。
输入格式:
输入包含若干组测试数据。每组数据的第1行给出两个正整数N (≤10)和L,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出N个以空格分隔的正整数,作为初始插入序列。最后L行,每行给出N个插入的元素,属于L个需要检查的序列。
简单起见,我们保证每个插入序列都是1到N的一个排列。当读到N为0时,标志输入结束,这组数据不要处理。
输出格式:
对每一组需要检查的序列,如果其生成的二叉搜索树跟对应的初始序列生成的一样,输出“Yes”,否则输出“No”。
输入样例:
4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0
输出样例:
Yes
No
No
在数据结构课程中何钦铭老师讲述了3中方法:
- 递归判断根节点相同,左子树相同,右子树相同;
- 建很多棵树,逐一去比较与第一棵树是否相同;
- 只建第一棵树T,其它序列按树T搜索,如果每次搜索经过的节点在前面出现过,则一致否则不一致。
这里采用第一种方法,代码量少。
def f(l1,l2):
if l1==[] and l2==[]:
return True
if l1[0]!=l2[0]:
return False
l1_left=[e for e in l1 if e>l1[0] ]
l1_right=[e for e in l1 if e<l1[0]]
l2_left=[e for e in l2 if e>l2[0]]
l2_right=[e for e in l2 if e<l2[0]]
return f(l1_left,l2_left) and f(l1_right,l2_right)
while True:
I=list(map(int ,input().split()))
if I[0]==0:
break;
[N,L]=I
link={}
for i in range(L+1):
link[i]=list(map(int,input().split()))
for i in range(1,L+1):
if f(link[0],link[i]):
print('Yes')
else:
print('No')