根据二叉树的前序遍历和中序遍历输出二叉树;假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列 {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; }
