Construct Binary Tree from Preorder and Inorder Traversal 结题报告
从前序遍历和中序遍历的结果重建一颗二叉树。
解题思路,随便写一个二叉树,然后写出前序和中序遍历的结果会发现特点。
二叉树的首节点必然是前序遍历的第一个节点,以这个节点在中序遍历的结果中作为划分,这个节点左侧的是左子树的节点,右侧是右子树节点。
例如,一个二叉树的前序遍历结果为:6 5 4 8 7 9
中序遍历结果为:4 5 6 7 8 9
前序遍历的第一个节点为6,就是这个二叉树的根节点。以6作为划分,中序遍历6以左的节点就是二叉树的左子树,以右的就是二叉树的右节点。按照此逻辑进行递归操作即可AC。
public class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder==null&&inorder==null||preorder.length==0) {
return null;
}
TreeNode root = new TreeNode(preorder[0]);
int i = 0;
for(i=0;i<inorder.length;i++) {
if(inorder[i]==preorder[0]) {
break;
}
}
int[] new_inorder_left, new_preorder_left, new_inorder_right, new_preorder_right;
if(i<inorder.length) {
//生成左子树时需要用到的inorder
new_inorder_left = new int[i];
System.arraycopy(inorder, 0, new_inorder_left, 0, i);
//生成左子树时需要用到的preorder
new_preorder_left = new int[i];
System.arraycopy(preorder, 1, new_preorder_left, 0, i);
root.left = buildTree(new_preorder_left, new_inorder_left);
//生成右子树时需要用到的inorder
new_inorder_right = new int[preorder.length - i - 1];
System.arraycopy(inorder, i + 1, new_inorder_right, 0, preorder.length - i - 1);
//生成右子树时需要用到的preorder
new_preorder_right = new int[preorder.length - i - 1];
System.arraycopy(preorder, 1 + i, new_preorder_right, 0, preorder.length - i - 1);
root.right = buildTree(new_preorder_right, new_inorder_right);
}
return root;
}
}