题目描述
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
其实现的二叉树为
问题分析
遍历前序遍历结果 ,3,在中序遍历找出 相同元素 ,中序遍历3左边的元素 【9】为3左子树的值,【15,20,7】为右子树中元素的值。
9,因为3的左子树就只有一个元素所以 3左子树的值就为9,
20,在中序中找出20,其左边区间只剩下【15】,右边为【7】,分析该问题可以递归实现。
代码实现
int index = 0;
public TreeNode buildTree(int[] preorder, int[] inorder) {
return biuldTreeHelper(preorder,inorder,0,preorder.length);
}
public TreeNode biuldTreeHelper(int[] preorder, int[] inorder, int left, int right) {
if(left >= right) {
return null;
}
if(index >= preorder.length) {
return null;
}
TreeNode treeNode = new TreeNode(preorder[index]);
int pos = find(preorder,inorder,index);
index++;
treeNode.left = biuldTreeHelper( preorder, inorder, left, pos);
treeNode.right = biuldTreeHelper( preorder, inorder, pos+1, right);
return treeNode;
}
public int find(int [] preorder,int [] inorder, int index) {
for(int i = 0; i < inorder.length; i ++) {
if(preorder[index] == inorder[i]) {
return i;
}
}
return -1;
}