重建二叉树--python

#4.重建2叉树
#输入二叉树的前序遍历和中序遍历的结果，重建出该二叉树
#前序遍历--根-->左-->右  中序遍历--左-->根-->右  后序遍历--左-->右-->根
#二叉树特例：①二叉搜索树--左子结点小于等于根结点，右子结点大于等于根结点②堆--分为最大堆和最小堆。在最大堆中根结点值最大，在最小堆中根结点值最小
class TreeNode:
	def __init__(self,x):
		self.val = x
		self.left = None
		self.right = None
# 前序遍历
pre = [1, 2, 4, 7, 3, 5, 6, 8]
#中序遍历
mid = [4, 7, 2, 1, 5, 3, 8, 6]
def findBinaryTree(pre,pre_start,pre_end,mid,mid_start,mid_end):
	if pre_start>pre_end or mid_start>mid_end or len(pre)==0 or len(mid) == 0:
		return None
	#前序遍历第一个为根结点
	root = TreeNode(pre[pre_start])
	#循环遍历中序遍历,分别找到对应的左右子树
	for i in range(mid_start,mid_end):
		#在中序遍历中找到根结点--区分左右子树
		if mid[i] == pre[pre_start]:
			root.left = findBinaryTree(pre,pre_start+1,i-mid_start+pre_start,mid,mid_start,i-1)
			root.right = findBinaryTree(pre,i-mid_start+pre_start+1,pre_end,mid,i+1,mid_end)
	return root
print(findBinaryTree(pre,0,7,mid,0,7).val)

def findBinaryTree1(pre,nid):
	if len(pre) == 0:
		return None
	if len(pre) == 1:
		return TreeNode(pre[0])
	else:
		root = TreeNode(pre[0])
		root.left = findBinaryTree1(pre[1:mid.index(pre[0])+1],mid[:mid.index(pre[0])])
		root.right = findBinaryTree1(pre[mid.index(pre[0])+1:],mid[mid.index(pre[0])+1:])
	return root