https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
这道题用递归就很明了了,每个子树,preorder的第一个数是根节点,然后在inorder里面找到这个根节点,inorder里左边的就是左子树,右边的是右子树,然后再递归构造左右子树。
这里有画图,讲得很清楚:http://fisherlei.blogspot.com/2013/01/leetcode-construct-binary-tree-from.html
另外,这个链接里还有人贴了iterative的解法。。。不过没看懂。。。。。。。。。。有空研究一下非递归的解法。。。。
public class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder == null || preorder.length == 0) return null;
TreeNode root = construct(preorder, 0, preorder.length-1, inorder, 0, inorder.length-1);
return root;
}
public TreeNode construct(int[] preorder, int pstart, int pend, int[] inorder, int istart, int iend){
if(pstart > pend) return null;
TreeNode root = new TreeNode(preorder[pstart]);
if(pstart == pend) return root;
int i;
for(i=istart; i<=iend; i++){
if(inorder[i]==preorder[pstart]) break;
}
root.left = construct(preorder, pstart+1, pstart+i-istart, inorder, istart, i-1);
root.right = construct(preorder, pstart+i-istart+1, pend, inorder, i+1, iend);
return root;
}
}
每个节点访问一遍,且每个节点都要进入下一层递归,即每个节点保存一遍,所以空间复杂度是O(n),n是总节点数。但构造每个节点时都要遍历一遍inorder数组来寻找根节点,所以worst case 是O(n^2)。worst case是只有左子树,没有右子树,这样,每次都要找到数组最后才能找到根节点。