牛客網(wǎng)劍指Offer——重建二叉樹

題目描述
輸入某二叉樹的前序遍歷和中序遍歷的結(jié)果,請重建出該二叉樹。假設(shè)輸入的前序遍歷和中序遍歷的結(jié)果中都不含重復(fù)的數(shù)字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6},則重建二叉樹并返回。

解題思路:
在二叉樹的前序遍歷序列中,第一個(gè)數(shù)字總是樹的根節(jié)點(diǎn)的值
在中序遍歷序列中,根節(jié)點(diǎn)的值在序列的中間。根據(jù)前序遍歷找到根節(jié)點(diǎn)后,中序遍歷序列中,位于根節(jié)點(diǎn)左邊的是左子樹的結(jié)點(diǎn)的值,位于根節(jié)點(diǎn)右邊的是右子樹的結(jié)點(diǎn)的值。

如下圖所示,前序遍歷序列的第一個(gè)數(shù)字1就是根結(jié)點(diǎn)的值。掃描中序遍歷序列,就能確定根結(jié)點(diǎn)的值的位置。根據(jù)中序遍歷特點(diǎn),在根結(jié)點(diǎn)的值1前面的3個(gè)數(shù)字都是左子樹結(jié)點(diǎn)的值,位于1后面的數(shù)字都是右子樹結(jié)點(diǎn)的值。

示例

同樣,在前序遍歷的序列中,根結(jié)點(diǎn)后面的3個(gè)數(shù)字就是3個(gè)左子樹結(jié)點(diǎn)的值,再后面的所有數(shù)字都是右子樹結(jié)點(diǎn)的值。這樣我們就在前序遍歷和中序遍歷兩個(gè)序列中,分別找到了左右子樹對應(yīng)的子序列。

既然我們已經(jīng)分別找到了左、右子樹的前序遍歷序列和中序遍歷序列,我們可以用同樣的方法分別去構(gòu)建左右子樹。也就是說,接下來的事情可以用遞歸的方法去完成。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if( pre.empty() )
            return NULL;
        TreeNode* root = new TreeNode( pre[0] );
        
        vector<int> left_pre, left_vin, right_pre, right_vin;
        
        //left_vin
        for( int i=0;vin[i]!=root->val;i++ )
            left_vin.push_back(vin[i]);
        //right_vin
        for( int i=left_vin.size()+1;i<vin.size();i++)
            right_vin.push_back(vin[i]);
        //left_pre
        for( int i=1;i<=left_vin.size();i++ )
            left_pre.push_back(pre[i]);
        //right_pre
        for( int i=left_vin.size()+1;i<pre.size();i++ )
            right_pre.push_back(pre[i]);
        
        root->left = reConstructBinaryTree( left_pre, left_vin );
        root->right = reConstructBinaryTree( right_pre, right_vin );
        
        return root;
    }
};
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • 樹 記錄《劍指offer》中所有關(guān)于樹的題目,以及LeetCode中的相似題目。 相關(guān)題目列表 題目 樹是一種最常...
    wenmingxing閱讀 1,517評論 2 13
  • 四、樹與二叉樹 1. 二叉樹的順序存儲(chǔ)結(jié)構(gòu) 二叉樹的順序存儲(chǔ)就是用數(shù)組存儲(chǔ)二叉樹。二叉樹的每個(gè)結(jié)點(diǎn)在順序存儲(chǔ)中都有...
    MinoyJet閱讀 1,741評論 0 7
  • 樹的概述 樹是一種非常常用的數(shù)據(jù)結(jié)構(gòu),樹與前面介紹的線性表,棧,隊(duì)列等線性結(jié)構(gòu)不同,樹是一種非線性結(jié)構(gòu) 1.樹的定...
    Jack921閱讀 4,764評論 1 31
  • 面試題7:重建二叉樹 題目: 輸入某二叉樹的前序遍歷和中序遍歷的結(jié)果。請重建該二叉樹。假設(shè)輸入的前序遍歷和中序遍歷...
    lyoungzzz閱讀 637評論 0 0
  • 前言 樹是數(shù)據(jù)結(jié)構(gòu)中的重中之重,尤其以各類二叉樹為學(xué)習(xí)的難點(diǎn)。一直以來,對于樹的掌握都是模棱兩可的狀態(tài),現(xiàn)在希望通...
    MrHorse1992閱讀 355,592評論 51 536

友情鏈接更多精彩內(nèi)容