重建二叉樹
1.題目
輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6},則重建二叉樹並返回。
2. 思路
這道題其實很簡單,思路比較清晰。
根據前序和中序構建。
1.在前序中確定根。參照物。
2.在中序中確定迭代的位置。以根劃分爲左右兩部分。
3.代碼以及分析
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre.length == 0 || pre.length != in.length){
return null;
}
TreeNode root = new TreeNode(pre[0]);//順序。
// root.val = pre[0];
for(int i = 0; i < pre.length; i++){
if(pre[0] == in[i]){
root.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, 1 + i ), Arrays.copyOfRange(in,0, i ));
root.right = reConstructBinaryTree(Arrays.copyOfRange(pre, 1 + i, pre.length), Arrays.copyOfRange(in, 1 + i, pre.length));
break;
}
}
return root;
}
}
- 新建TreeNode 時就要直接給他初始值。(看情況)
- 當使用了
TreeNode root = new TreeNode()
root.val = pre[0];
就不可以通過。
不可以用root.val 再初始化。(???這一點存有疑惑,歡迎指教)
解答: 因爲已經給出了有參構造器。此時在構造對象時如果沒有提供參數就會被視爲不合法。 - break!找到根就跳出來!
- 一定要考慮邊界條件。什麼時候不符合條件。(數組長度爲0,以及,前序和中序的長度不相等)
- 要注意基礎函數的使用,Arrays.copyOfRange(要拷貝的數組名 , 起始位置 , 結束爲止‘當前索引+1’ )。