LeetCood:重建二叉樹

輸入某二叉樹的前序遍歷和中序遍歷的結(jié)果但惶,請重建該二叉樹阴汇。假設(shè)輸入的前序遍歷和中序遍歷的結(jié)果中都不含重復(fù)的數(shù)字页响。

例如非洲,給出

前序遍歷 preorder = [3,9,20,15,7]
中序遍歷 inorder = [9,3,15,20,7]

返回如下的二叉樹:

    3
   / \
  9  20
    /  \
   15   7

限制:

0 <= 節(jié)點個數(shù) <= 5000

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof


思路:

首先如果要構(gòu)建一棵樹我們是否要找到樹的根節(jié)點然后找到左子樹、右子樹俯画,然后再從左析桥、右子樹分別找到根節(jié)點,再分為左右子樹艰垂,一直重復(fù)這個行為直到?jīng)]有元素為止泡仗,這就是分治的思想,將大的問題分為小的相同的問題猜憎。再總結(jié)一下前序遍歷和中序遍歷的特點娩怎,在前序遍歷中根節(jié)點肯定是排在樹(完整序列)的首位或者子樹(子序列)的首位,而中序遍歷根節(jié)點肯定是排在樹(完整序列)的中間或者子樹(子序列)的中間胰柑,這樣得出前序遍歷可以方便為我們找到根節(jié)點截亦,而中序遍歷方便為我們找到左爬泥、右子樹,然后再從左子樹崩瓤,右子樹中繼續(xù)上述操作袍啡,我們就遞歸構(gòu)建出一顆樹了


代碼題解:

#include<iostream>
#include <unordered_map>
#include<vector>
using namespace std;
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
//中序遍歷值和索引的映射關(guān)系
unordered_map<int, int> index;

/*
p_left 前序遍歷(子)序列的首元素谷遂,p_right 前序遍歷的尾元素葬馋,i_left 中序遍歷的首元素,i_right 中序遍歷的尾元素
*/
TreeNode* recursiveTree(vector<int>& preorder, vector<int>& inorder, int p_left, int p_right, int i_left, int i_right) {
    if (p_left > p_right)
    {
        return nullptr;
    }
    int root_value = preorder[p_left];//根的值一定是前序遍歷子)序列的首元素
    int root_index = index[root_value]; //根據(jù)根的值我們找到根在中序遍歷的索引肾扰,那么索引左側(cè)則為左子樹畴嘶,右側(cè)則為右子樹
    TreeNode * root = new TreeNode(root_value); //new一個當(dāng)前找到的節(jié)點
    int size_left_tree = root_index - i_left; // 根在中序遍歷的索引減去中序遍歷的首元素的位置 = 左子樹的大小
    root->left = recursiveTree(preorder, inorder, p_left + 1, p_left + size_left_tree, i_left, root_index - 1);//遞歸左子樹,并讓當(dāng)前節(jié)點的左子樹指向遞歸結(jié)果
    root->right = recursiveTree(preorder, inorder, p_left + size_left_tree + 1, p_right, root_index + 1, i_right);//遞歸右子樹
    return root;
}

/*
構(gòu)建樹
*/
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
    for (int i = 0; i < inorder.size(); i++)
    {
        index[inorder[i]] = i;
    }
    return recursiveTree(preorder, inorder, 0, preorder.size() - 1, 0, inorder.size() - 1);
}
int main() {
    //自行添加測試用例
    
    return 0;
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末集晚,一起剝皮案震驚了整個濱河市窗悯,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌偷拔,老刑警劉巖蒋院,帶你破解...
    沈念sama閱讀 221,331評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異莲绰,居然都是意外死亡欺旧,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,372評論 3 398
  • 文/潘曉璐 我一進(jìn)店門蛤签,熙熙樓的掌柜王于貴愁眉苦臉地迎上來辞友,“玉大人,你說我怎么就攤上這事震肮〕屏” “怎么了?”我有些...
    開封第一講書人閱讀 167,755評論 0 360
  • 文/不壞的土叔 我叫張陵戳晌,是天一觀的道長鲫尊。 經(jīng)常有香客問我,道長沦偎,這世上最難降的妖魔是什么疫向? 我笑而不...
    開封第一講書人閱讀 59,528評論 1 296
  • 正文 為了忘掉前任,我火速辦了婚禮豪嚎,結(jié)果婚禮上鸿捧,老公的妹妹穿的比我還像新娘。我一直安慰自己疙渣,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 68,526評論 6 397
  • 文/花漫 我一把揭開白布堆巧。 她就那樣靜靜地躺著妄荔,像睡著了一般泼菌。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上啦租,一...
    開封第一講書人閱讀 52,166評論 1 308
  • 那天哗伯,我揣著相機(jī)與錄音,去河邊找鬼篷角。 笑死焊刹,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的恳蹲。 我是一名探鬼主播虐块,決...
    沈念sama閱讀 40,768評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼嘉蕾!你這毒婦竟也來了贺奠?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,664評論 0 276
  • 序言:老撾萬榮一對情侶失蹤错忱,失蹤者是張志新(化名)和其女友劉穎儡率,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體以清,經(jīng)...
    沈念sama閱讀 46,205評論 1 319
  • 正文 獨居荒郊野嶺守林人離奇死亡儿普,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,290評論 3 340
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了掷倔。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片眉孩。...
    茶點故事閱讀 40,435評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖今魔,靈堂內(nèi)的尸體忽然破棺而出勺像,到底是詐尸還是另有隱情,我是刑警寧澤错森,帶...
    沈念sama閱讀 36,126評論 5 349
  • 正文 年R本政府宣布吟宦,位于F島的核電站烹笔,受9級特大地震影響扛邑,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜导街,卻給世界環(huán)境...
    茶點故事閱讀 41,804評論 3 333
  • 文/蒙蒙 一瓦阐、第九天 我趴在偏房一處隱蔽的房頂上張望蜗侈。 院中可真熱鬧,春花似錦睡蟋、人聲如沸踏幻。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,276評論 0 23
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽该面。三九已至夭苗,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間隔缀,已是汗流浹背题造。 一陣腳步聲響...
    開封第一講書人閱讀 33,393評論 1 272
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點兒被人妖公主榨干…… 1. 我叫王不留猾瘸,地道東北人界赔。 一個月前我還...
    沈念sama閱讀 48,818評論 3 376
  • 正文 我出身青樓,卻偏偏與公主長得像牵触,于是被迫代替她去往敵國和親淮悼。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 45,442評論 2 359

推薦閱讀更多精彩內(nèi)容