题目描述
题目来源:牛客-剑指offer
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
分析
根据前序遍历和中序遍历的特点,前序遍历的第一个元素即根节点,在中序序列中该节点的左边为左子树,右边为右子树。
Java实现
public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
return reConstructBinaryTree(pre, 0, pre.length - 1 , in, 0, in.length - 1 );
}
private static TreeNode reConstructBinaryTree(int[] pre , int preLow, int preHigh, int[] in, int inLow,int inHigh){
if(preLow > preHigh || inLow > inHigh) return null;
TreeNode treeNode = new TreeNode(pre[preLow]); //根节点为pre的第一个元素
for (int i = 0; i < in.length; i++) {
//找到pre[lowHigh]在in序列中的索引(即根节点在in序列中位置,根节点左边为左子树节点,右边为右子树节点)
if( pre[preLow] == in[i] ){//startPre+i-startIn i-startIn+startPre+1
treeNode.left = reConstructBinaryTree(pre, preLow + 1 , preLow + i - inLow, in, inLow , i - 1);
treeNode.right = reConstructBinaryTree(pre, i - inLow + preLow + 1, preHigh, in, i + 1, inHigh);
break;
}
}
return treeNode;
}