https://oj.leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
Given preorder and inorder traversal of a tree, construct the binary tree.
Note:
You may assume that duplicates do not exist in the tree.
public TreeNode buildTree(int[] preorder, int[] inorder)
这一题利用到了一个特性是:一棵树的左子树的节点是不变的.....嗯,我在说什么呢?
其实是这样的:虽然preoreder和inorder的顺序肯定是不一样的。但是根节点左子树的范围和右边的范围是一样的。毕竟不管是in order还是pre order,都是先跑完左子树 + 根节点 再跑右子树的。所以如果说如果左子树的节点数目加上根节点的数目是i,那么不管是preorder还是inorder数组,前面i个节点都表示了左子树+根节点。
然后我们基于这个特性,再找到preorder和inorder的不同点。 preorder的第一个节点就是根节点,inorder要跑完所有左子树的节点才跑根节点。根据以上两个特质,做法如下:
1. 选取当前preorder的第一个元素(或者子集里的第一个元素)。
2.找到这个元素在inorder里对应的位置。譬如说这个位置是 i, 那么前面i - 1个位置就是这棵树的左子树了。同样的,preorder 在第一个元素的后面的i - 1个位置也都是左子树的内容。
3. 然后在inorder里以这个i为界,作为当前的根节点,然后分成左右两个部分,分别就是左子树和右子树的内容。同样,我们也可以根据preorder的特性找到左子树和右子树在数组里的范围。
4. 以上面3步找到的左子树和右子树在preorder和inorder里的范围,向左右边进行递归,构建左右子树。每一个递归由1开始继续做继续递归。
5。假设数组子集的范围由head和tail表示。一旦pre_head/in_head > pre_tail/in_tail(两者都可以),就表示该子树为空。返回null, 一旦pre_head/in_head == pre_tail/in_tail
表示的就是当前子树为单节点,返回new TreeNode(preorder[pre_start])即可。
给出代码:
public TreeNode buildTree(int[] preorder, int[] inorder) {
return helper(0, preorder.length - 1, 0, inorder.length - 1, preorder, inorder);
}
public TreeNode helper(int pre_start, int pre_end, int in_start, int in_end, int[] preorder, int[] inorder){
if(in_start > in_end)
return null;
if(in_start == in_end)
return new TreeNode(inorder[in_start]);
int in_pos = 0;
for(int i = in_start; i <= in_end; i++){
if(preorder[pre_start] == inorder[i]){
in_pos = i;
break;
}
}
TreeNode root = new TreeNode(preorder[pre_start]);
root.left = helper(pre_start + 1, pre_start + in_pos - in_start, in_start, in_pos - 1,preorder, inorder);
root.right = helper(pre_start + in_pos - in_start + 1, pre_end, in_pos + 1, in_end, preorder, inorder);
return root;
}