输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
解法1:递归思路。
在前序遍历中,根节点即为第一个数。
确定根节点后,在中序遍历中找到根节点对应位置,根据中序遍历特点,根节点位置左侧属于左子树,右侧属于为右子树。
这样就确定了左子树的节点个数。根据节点个数在前序遍历中划分出左子树区间。
形成循环节
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
return buildTreePart(preorder, 0, preorder.length, inorder, 0, inorder.length);
}
// 左闭右开
private TreeNode buildTreePart(int[] preorder, int pre_left, int pre_right,
int[] inorder, int in_left, int in_right)
{
if(pre_right-pre_left <= 0) return null;
if(pre_right-pre_left==1) return new TreeNode(preorder[pre_left]);
TreeNode root = new TreeNode(preorder[pre_left]);
for(int i=in_left;i<in_right;i++)
{
// pre_left为子数的根节点,在中序遍历中找出根节点位置
if(preorder[pre_left] == inorder[i])
{
int len = i-in_left; //计算左子树元素个数
//创建左子树
root.left = buildTreePart(preorder, pre_left+1, pre_left+len+1,
inorder, in_left, i);
//创建右子树
root.right = buildTreePart(preorder, pre_left+len+1, pre_right,
inorder, i+1, in_right);
break;
}
}
return root;
}
}