《劍指offer》c++版本 7.重建二叉樹

 如題:

輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和
中序遍歷的結果中都不含重複的數字。例如輸入前序遍歷序列{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),感興趣的同學可以加羣討論、交流、資料查找等,前進的道路上,你不是一個人奧^_^。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章