根据一棵树的前序遍历与中序遍历构造二叉树
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
int preIndex = 0;
public TreeNode2 buildTreeChild(int[] preorder, int[] inorder,int inbegin,int inend) {
//判断是否有左树或者是右树
if(inbegin > inend) {
return null;
}
//找到前序遍历的根结点root
TreeNode2 root = new TreeNode2(preorder[preIndex]);
//找root在中序遍历的下标
int rootIndex = findIndexOfInorder(inorder,inbegin,inend,preorder[preIndex]);
//前序遍历 ++
preIndex++;
root.left = buildTreeChild(preorder ,inorder,inbegin, rootIndex-1);
root.right = buildTreeChild(preorder ,inorder,rootIndex+1,inend);
return root;
}
public int findIndexOfInorder(int[] inorder,int inbegin, int inend,int val){
for(int i = inbegin;i <= inend;i++) {
if(inorder[i] == val) {
return i;
}
}
return -1;
}
public TreeNode2 buildTree2(int[] preorder, int[] inorder) {
if(preorder == null || inorder == null) {
return null;
}
if(preorder.length == 0 || inorder.length ==0) {
return null;
}
return buildTreeChild(preorder,inorder,0,inorder.length-1);
}