这道题是通过preorder和inorder的顺序来确认整个tree。
思路:
preorder: root, left, right; inorder: left, root, right。
example: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7];
preorder第一个一定是root,记录下root的值,在inorder里面找到root对应的index,那么root的左边就是左子树,root的右边就是右子树。recurively找到root和左右子树。
难点在于确认preorder里面左右的大小,需要通过inorder里面算出的index确认。
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
return dfs(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1);
}
public TreeNode dfs(int[] preorder, int pleft, int pright,
int[] inorder, int ileft, int iright){
if(ileft > iright || pleft > pright) return null;
TreeNode root = new TreeNode(preorder[pleft]);
int i = 0;
for(i = ileft; i < inorder.length; i++){
if(root.val == inorder[i]) break;
}
root.left = dfs(preorder, pleft + 1,pleft + i - ileft, inorder, ileft, i - 1);
root.right = dfs(preorder, pleft + i - ileft + 1, pright, inorder, i + 1, iright);
return root;
}
}
Reference:
106. Construct Binary Tree from Inorder and Postorder Traversal