题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
知识点
二叉树:
二叉树是n(n>=0)个结点的有限集合,该集合或者为空集(称为空二叉树),或者由一个根结点和两棵互不相交的、分别称为根结点的左子树和右子树组成。
二叉树的三种遍历方式:
- 前序遍历:先访问根节点,再访问左节点,最后访问右节点。
- 中序遍历:先访问左节点,再访问根节点,最后访问右节点。
- 后序遍历:先访问左节点,再访问右节点,最后访问根节点。
思路
考查的是拆分问题的能力,由前序遍历和中序遍历的性质可得:
前序的第一个为根节点,可以将中序数组拆分为左子树和右子树;
以此类推,可以通过递归解出结果。
代码
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre.length==0||in.length==0){
return null;
}
TreeNode root = new TreeNode(pre[0]);
int j=0;
while(in[j]!=pre[0]) j++;
root.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,j+1),Arrays.copyOfRange(in,0,j));
root.right = reConstructBinaryTree(Arrays.copyOfRange(pre,j+1,pre.length),Arrays.copyOfRange(in,j+1,in.length));
return root;
}
}