根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
思路与利用后序遍历和中序遍历构造二叉树一样,先从前序遍历中找出二叉树的根节点,前序遍历的第一个元素就是二叉树的根节点,然后在中序遍历中找出根结点的位置,就可以得知左子树和右子树的长度,然后我们就得到了根节点以及左子树和右子树的前序遍历和中序遍历数组。按照这个思路进行递归就可以构造出一颗完整的二叉树。
代码:
public class BuildTree {
private static class TreeNode{
int val;
TreeNode left;
TreeNode right;
TreeNode(int val){
this.val=val;
}
}
private static TreeNode solve(int[] preorder,int[] inorder) {
return search(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
}
private static TreeNode search(int[] preorder,int prestart,int preend,int[] inorder,int instart,int inend) {
if(preend<prestart) return null;
TreeNode root=new TreeNode(preorder[prestart]);
int position=searchPosition(inorder,instart,inend,preorder[prestart]);
root.left=search(preorder,prestart+1,prestart+(position-instart),inorder,instart,position-1);
root.right=search(preorder,prestart+(position-instart)+1,preend,inorder,position+1,inend);
return root;
}
private static int searchPosition(int[] inorder,int instart,int inend,int key) {
for(int i=instart;i<=inend;i++) {
if(key==inorder[i]) return i;
}
return -1;
}
}