如題:
輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和
中序遍歷的結果中都不含重複的數字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序
遍歷序列{4,7,2,1,5,3,8,6},則重建二叉樹並返回。
根據前序和中序數列重建二叉樹,這道題,還有一道類似題,根據後續和中序數列重建二叉樹。兩題解法完全一樣,僅僅方向不同。這是學習二叉樹的時候必做的題型,和遍歷二叉樹性質類似。
看題,我們知道,前序遍歷得到的數列,依次是各子樹的根節點,如根節點,根節點左子樹,根節點右子樹......,此外,中序遍歷數列中,根節點的值位於中間,左右兩邊依次是左右子樹。結合前序數列性質,我們可以這樣做,先再前序數列求得根節點的值,然後,查找中序遍歷,此時,中序數列中,位於根節點左邊的數列即是左子樹,右邊的爲右子樹。先後求得根節點的左子樹以及右子樹。之後,採用遞歸,依次求得節點的左右子樹即可。
後序遍歷也是一樣處理,只不過方向變反。後序遍歷數列,從後往前依次是根節點,根節點右子樹,根節點左子樹,即左右子樹求取順序和上述題型相反。
下面是本題的c++解法:
//模板題,先計算根節點,然後分別計算左子樹和右子樹的根節點
class Solution {
public:
TreeNode *constructTree(vector<int> pre,int *preStart, int preEnd, vector<int> vin, int vinStart, int vinEnd){
int i;
TreeNode *root;
//特殊情況處理
if (vinStart > vinEnd) return NULL;
root = new TreeNode(pre[*preStart]);
//找到左子樹和右子樹的範圍
for (i = vinStart; i <= vinEnd; i++){
if (pre[*preStart] == vin[i])
break;
}
//指向下一個根節點
*preStart = *preStart + 1;
root->left = constructTree(pre, preStart, preEnd, vin,vinStart,i-1);
root->right = constructTree(pre, preStart, preEnd, vin,i+1, vinEnd);
return root;
}
TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin) {
int preStart = 0;
if (pre.empty()) return NULL;
return constructTree(pre, &preStart, pre.size()-1,vin,0,vin.size()-1);
}
};
=============================================================================================
Linux應用程序、內核、驅動、後臺開發交流討論羣(745510310),感興趣的同學可以加羣討論、交流、資料查找等,前進的道路上,你不是一個人奧^_^。