输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/
9 20
/
15 7
限制:
0 <= 节点个数 <= 5000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
前序遍历和中序遍历的特点得知:
前序遍历的根节点始终在第一位,从而根据已知的根节点的值找到中序遍历的根节点的值,进而得出左子树和右子树的长度,递归分而治之。
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder == null || inorder == null || preorder.length == 0 || inorder.length == 0)
return null;
return buildTree(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
}
TreeNode buildTree(int[] preorder,int lpre,int hpre,int[] inorder,int lin,int hin) {
if(lin > hin || lpre > hpre)return null;
TreeNode root = new TreeNode(preorder[lpre]);
int index = lin;
while(inorder[index] != root.val)index++;
int left_len = index - lin;
//注意边界条件,尤其是左子树的起始与结束
root.left = buildTree(preorder,lpre+1,lpre+left_len,inorder,lin,lin + left_len);
root.right = buildTree(preorder,lpre+1+left_len,hpre,inorder,index+1,hin);
return root;
}