题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路:
先序遍历第一个位置肯定是根节点node,
中序遍历的根节点位置在中间p,在p左边的肯定是node的左子树的中序数组,p右边的肯定是node的右子树的中序数组
另一方面,先序遍历的第二个位置到p,也是node左子树的先序子数组,剩下p右边的就是node的右子树的先序子数组
把四个数组找出来,分左右递归调用即可!
/** * 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) { return reConBTree(pre,0,pre.length-1,in,0,in.length-1); } public TreeNode reConBTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn){ if(startPre>endPre||startIn>endIn){//到达边界条件 return null; } //新建一个TreeNode TreeNode root=new TreeNode(pre[startPre]); //遍历中序数组 for(int i=startIn;i<=endIn;i++){ if(pre[startPre]==in[i]){ //重构左子树 root.left=reConBTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1); //重构右子树 root.right=reConBTree(pre,startPre+i-startIn+1,endPre,in,i+1,endIn); } } return root; } }