根据二叉树的前序遍历和中序遍历输出二叉树;假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列 {1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
这是一道《剑指Offer》上的题目,主要考察了二叉树的遍历方法;
刚开始做这道题,首先得掌握二叉树的三种遍历方式:
前序遍历:先访问根结点,然后前序遍历左子树,再前序遍历右子树。
中序遍历:中序遍历根结点的左子树,然后访问根结点,最后中序遍历右子树。
后序遍历:后序遍历根结点的左子树,再后序遍历根结点的右子树,最后访问根结点。
有了基础概念之后,就可以分析这道题了。既然要用递归,那就得先明白三点: 1,需要干什么(输出一个二叉树);2,什么步骤(每次都需要先根据前序遍历的根节点来截取中序遍历,将其作为根节点的左节点和右节点);3,什么时候结束(当最后的数组长度为1,而树的结束也可能到只有左节点和只有右节点)
那么根据这三点就能大致写出递归方法了;
需要注意的是,这里截取数组的时候需要知道数组的下标,所以先写了个方法获取数组下标:
private static int getIndex(int[] arr, int val) {
for (int i = 0 ; i < arr.length ; i++ ) {
if (val == arr[i]) {
return i;
}
}
return -1;
}
然后就是构建树的递归方法:
/**
* 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。
* 每次都需要先根据前序遍历的根节点来截取中序遍历,将其作为根节点的左节点和右节点
* @param DLRArr 前序
* @param LDRArr 中序
* @param root 前序遍历的第一个值即为根节点
* @return
*/
public static TreeNode geneTree(int[] DLRArr, int [] LDRArr, int root) {
TreeNode treeNode = new TreeNode(root);
if(DLRArr.length == 1) { // 叶子结点
return treeNode;
}
int midIndex = getIndex(LDRArr, DLRArr[0]);
if (0 == midIndex) { // 左节点为空
treeNode.leftNode = null;
treeNode.rightNode = geneTree(Arrays.copyOfRange(DLRArr, midIndex + 1, DLRArr.length), Arrays.copyOfRange(LDRArr, midIndex + 1, LDRArr.length), Arrays.copyOfRange(DLRArr, midIndex + 1, DLRArr.length)[0]);
return treeNode;
}
if (midIndex == LDRArr.length -1) { // 右节点为空
treeNode.rightNode = null;
treeNode.leftNode = geneTree(Arrays.copyOfRange(DLRArr, 1, midIndex + 1), Arrays.copyOf(LDRArr, midIndex ), Arrays.copyOfRange(DLRArr, 1, midIndex + 1)[0]);
return treeNode;
}
treeNode.leftNode = geneTree(Arrays.copyOfRange(DLRArr, 1, midIndex + 1), Arrays.copyOf(LDRArr, midIndex ), Arrays.copyOfRange(DLRArr, 1, midIndex + 1)[0]);
treeNode.rightNode = geneTree(Arrays.copyOfRange(DLRArr, midIndex + 1, DLRArr.length), Arrays.copyOfRange(LDRArr, midIndex + 1, LDRArr.length), Arrays.copyOfRange(DLRArr, midIndex + 1, DLRArr.length)[0]);
return treeNode;
}