重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
二叉树前序遍历规则为:根节点->左子树->右子树,简记为根左右,而中序遍历规则为:左子树->根节点->右子树,可简记为左根右。
而且可以知道的是给定了我们前序遍历和中序遍历我们是可以求出中序遍历的。
前序遍历的第一个节点肯定为根节点,这个节点在中序遍历的序列中处于中间节点,在中序遍历中找到根节点的位置,在他左边的区间就是他的左子树,右边的区间就是他的右子树。可根据此来判断前序节点中根节点下一个节点的是根节点的左子树根还是右子树根。重复递归下去即可。
重建代码如下
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
# 返回构造的TreeNode根节点
def reConstructBinaryTree(self, pre, tin):
# write code here
if len(pre) == 0:
return None
if len(pre) == 1:
return TreeNode(pre[0])
else:
flag = TreeNode(pre[0])
flag.left = self.reConstructBinaryTree(
pre[1:tin.index(flag.val) + 1], tin[:tin.index(pre[0])])
flag.right = self.reConstructBinaryTree(
pre[tin.index(flag.val) + 1:], tin[tin.index(pre[0]) + 1:])
return flag
def main():
pass
if __name__ == '__main__':
main()