算法题目-二叉树的重建

来源:互联网 发布:9.3越狱软件源 编辑:程序博客网 时间:2024/06/03 15:02

题目:

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

分析:

对于二叉树,前序遍历是,根左右。{1,2,4,7,3,5,6,8}

中序遍历是,左根右。{4,7,2,1,5,3,8,6}

因此可知前序遍历中的第一个数1为根节点,再看中序遍历,可以得到1之前的为左子树{4,7,2},1之后的为右子树{5,3,8,6}

再继续进行递归分析即可。

解答:

/** * Definition for binary tree * public class TreeNode { *     int val; *     TreeNode left; *     TreeNode right; *     TreeNode(int x) { val = x; } * } */public class Solution {    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {        if(pre.length==0||in.length==0||pre.length!=in.length){            return null;        }        int i=0;        TreeNode root=new TreeNode(pre[0]);        while(in[i]!=root.val){            i++;        }        int[] preLeft=new int[i];        int[] inLeft=new int[i];        int[] preRight=new int[pre.length-1-i];        int[] inRight=new int[in.length-1-i];        for(int j=0;j<pre.length;j++){            if(j<i){                preLeft[j]=pre[j+1];                inLeft[j]=in[j];            }else if(j>i){                preRight[j-i-1]=pre[j];                inRight[j-i-1]=in[j];            }        }        root.left=reConstructBinaryTree(preLeft,inLeft);        root.right=reConstructBinaryTree(preRight,inRight);        return root;    }}



原创粉丝点击