根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7] 中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3 / \ 9 20 / \ 15 7
分析:先序遍历的顺序是(根-左子树-右子树),中序遍历的顺序是(左子树-根-右子树),所以先序遍历输出的第一个节点一定是根节点,从而分别递归构造子树。
/**
* 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) {
int length1=preorder.length-1;
int length2=inorder.length-1;
return rebuildTree(preorder,0,length1,inorder,0,length2);
}
public TreeNode rebuildTree(int[] preorder,int start1,int end1,int[] inorder,int start2,int end2){
if(start1>end1||start2>end2)
return null;
TreeNode root=new TreeNode(preorder[start1]);
for(int i=start2;i<=end2;i++){
if(inorder[i]==preorder[start1]){
root.left=rebuildTree(preorder,start1+1,start1+i-start2,inorder,start2,i-1);
root.right=rebuildTree(preorder,start1+i-start2+1,end1,inorder,i+1,end2);
break;
}
}
return root;
}
}