判断二叉搜索树是否同构

原题来源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中方法:

  1. 递归判断根节点相同,左子树相同,右子树相同;
  2. 建很多棵树,逐一去比较与第一棵树是否相同;
  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')
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值