【leetcode】105.(Hard)Construct Binary Tree from Preorder and Inorder Traveral
阿新 • • 發佈:2018-12-27
解題思路:
遞迴
先序遍歷的第一個結點是根節點
根據根節點和中序遍歷的結果可以找到左子樹的結點和右子樹的結點。
繼續遞迴,將左子樹的結點和右子樹的結點構造成樹,最後一層層創接起來。
時間複雜度:O(nlogn) 遞迴每層的耗時為O(n),共logn層
空間複雜度:O(nlogn) 遞迴每層所佔空間為O(n),共logn層
如果將函式寫成指標的形式(即不新建左子樹和右子樹的陣列,直接在原陣列中指明左子樹和右子樹的範圍),時間複雜度可提升到O(logn)
提交程式碼:
class Solution {
public TreeNode buildTree(int[ ] preorder, int[] inorder) {
if(preorder.length==0) return null;
if(preorder.length==1) {
TreeNode node=new TreeNode(preorder[0]);
return node;
}
TreeNode root=new TreeNode(preorder[0]);
int i,j,leftSize=0;
for(i=0;i<inorder.length&& inorder[i]!=root.val;i++)
leftSize++;
int[] leftInOrder=new int[leftSize];
int[] leftPreOrder=new int[leftSize];
int[] rightInOrder=new int[inorder.length-leftSize-1];
int[] rightPreOrder=new int[inorder.length-leftSize-1];
for(i=0;i< leftSize;i++)
leftInOrder[i]=inorder[i];
for(i=leftSize+1,j=0;i<inorder.length;i++,j++)
rightInOrder[j]=inorder[i];
for(i=0;i<leftSize;i++)
leftPreOrder[i]=preorder[i+1];
for(i=leftSize+1,j=0;i<preorder.length;i++,j++)
rightPreOrder[j]=preorder[i];
root.left=buildTree(leftPreOrder,leftInOrder);
root.right=buildTree(rightPreOrder,rightInOrder);
return root;
}
}
執行結果: