7、重建二叉树

别来无恙 提交于 2020-02-17 02:13:41

重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

思路分析

前序遍历的确定根节点,可以根据此点分别找到左子树跟右子树的前序遍历序列和中序遍历序列,通过递归再分别构建子树的左子树和右子树。代码如下:

public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
	return reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
}
public TreeNode reConstructBinaryTree(int[] pre,int preStart,int preEnd,int[] in,int inStart,int inEnd){
    if(preStart>preEnd||inStart>inEnd)return null;
    TreeNode root=new TreeNode(pre[preStart]);
    for(int i=inStart;i<=inEnd;i++){
        if(in[i]==pre[preStart]){
            root.left=reConstructBinaryTree(pre,preStart+1,preStart+i-inStart,in,inStart,i-1);
            root.right=reConstructBinaryTree(pre,preStart+i-inStart+1,preEnd,in,i+1,inEnd);
        }
    }
    return root;
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!