面試題7:根據(jù)前序遍歷和中序遍歷重建二叉樹

假設(shè)輸入的前序遍歷和中序遍歷的結(jié)果都不含重復(fù)的數(shù)字哩照,重建二叉樹并輸出根節(jié)點(diǎn)架谎。二叉樹的定義如下:

struct BTreeNode
{
    int        value;
    BTreeNode* pLeft;
    BTreeNode* pRight;
};

解析:這道題看起來挺難矩欠,但其實(shí)可以采用遞歸的方法卑雁,找到規(guī)律了就很好解赶撰。前序遍歷序列的第一個(gè)節(jié)點(diǎn)即為整個(gè)樹的根贮尉,后面的序列的左半部分是左子樹琼了,而且遵循一樣的規(guī)則嫡丙;右半部分是右子樹莹汤,也同樣遵循一樣的規(guī)則快鱼。中序遍歷序列在中間可以找到根節(jié)點(diǎn),這個(gè)節(jié)點(diǎn)左邊的部分序列即為左子樹的序列纲岭,右邊部分的序列即為右子樹的序列攒巍。根據(jù)中序遍歷序列的左子樹序列長度可以在前序遍歷序列中找到左子樹和右子樹的分界點(diǎn)。找到了這個(gè)規(guī)律荒勇,寫出代碼就很容易了柒莉。

代碼肯定采用遞歸的寫法。首先從前序序列中讀取第一個(gè)節(jié)點(diǎn)沽翔,即為當(dāng)前樹的根節(jié)點(diǎn)兢孝,構(gòu)建這個(gè)根節(jié)點(diǎn)。然后在中序序列中找到這個(gè)根節(jié)點(diǎn)仅偎,該節(jié)點(diǎn)的左半部分即為左子樹的中序遍歷序列跨蟹,該序列的長度即為左子樹的節(jié)點(diǎn)個(gè)數(shù)。從前序序列的第二個(gè)開始橘沥,根據(jù)左子樹節(jié)點(diǎn)個(gè)數(shù)即可推斷出剩余的前序序列中多少屬于左子樹窗轩,剩下的屬于右子樹。至于邊界情況座咆,當(dāng)前序序列和中序序列都只剩一個(gè)節(jié)點(diǎn)痢艺,那就說明構(gòu)建快要完成了,如果判斷前序序列的這個(gè)節(jié)點(diǎn)和中序序列的這個(gè)節(jié)點(diǎn)是同一個(gè)節(jié)點(diǎn)介陶,那就說明輸入的遍歷序列合法堤舒。


答案:

BTreeNode* ConstructCore
    (int* startPreOrder, int* endPreOrder, int* startInOrder, int* endInOrder)
{
    //construct the root
    auto root = new BTreeNode();
    root->pLeft = root->pRight = nullptr;
    root->value = *startPreOrder;

    //there's only one node left
    if (startPreOrder == endPreOrder)
    {
        if (startInOrder==endInOrder && *startPreOrder==*startInOrder)
            return root;
        else throw exception();
    }

    //find the root in the InOrder sequence
    int *rootInOrder = startInOrder;
    while (*rootInOrder != *startPreOrder && rootInOrder<=endInOrder)
        ++rootInOrder;
    if (rootInOrder==endInOrder && *rootInOrder!=*startPreOrder)
        throw exception();

    //mark left subtree and right subtree
    int *startLeftTreePreOrder = startPreOrder+1;
    int *endLeftTreePreOrder = startPreOrder+(rootInOrder-startInOrder);
    int *startLeftTreeInOrder = startInOrder;
    int *endLeftTreeInOrder = rootInOrder-1;

    int *startRightTreePreOrder = endLeftTreePreOrder+1;
    int *endRightTreePreOrder = endPreOrder;
    int *startRightTreeInOrder = rootInOrder+1;
    int *endRightTreeInOrder = endInOrder;

    //recursively construct left subtrees and right subtrees
    if (*rootInOrder == *startPreOrder && rootInOrder>startInOrder)
        root->pLeft = 
                ConstructCore(startLeftTreePreOrder, endLeftTreePreOrder, 
                        startLeftTreeInOrder, endLeftTreeInOrder);
    if (*rootInOrder == *startPreOrder && rootInOrder<endInOrder)
        root->pRight = 
                ConstructCore(startRightTreePreOrder, endRightTreePreOrder,
                        startRightTreeInOrder, endRightTreeInOrder);

    return root;
}

BTreeNode* Construct(int* preOrder, int* inOrder, int length)
{
    if (preOrder == nullptr || inOrder == nullptr || length <= 0)
        return nullptr;

    return ConstructCore(preOrder, preOrder+length-1, 
            inOrder, inOrder+length-1);
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市哺呜,隨后出現(xiàn)的幾起案子舌缤,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 206,214評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件国撵,死亡現(xiàn)場離奇詭異陵吸,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)介牙,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,307評論 2 382
  • 文/潘曉璐 我一進(jìn)店門走越,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人耻瑟,你說我怎么就攤上這事旨指。” “怎么了喳整?”我有些...
    開封第一講書人閱讀 152,543評論 0 341
  • 文/不壞的土叔 我叫張陵谆构,是天一觀的道長。 經(jīng)常有香客問我框都,道長搬素,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 55,221評論 1 279
  • 正文 為了忘掉前任魏保,我火速辦了婚禮熬尺,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘谓罗。我一直安慰自己粱哼,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,224評論 5 371
  • 文/花漫 我一把揭開白布檩咱。 她就那樣靜靜地躺著揭措,像睡著了一般。 火紅的嫁衣襯著肌膚如雪刻蚯。 梳的紋絲不亂的頭發(fā)上绊含,一...
    開封第一講書人閱讀 49,007評論 1 284
  • 那天,我揣著相機(jī)與錄音炊汹,去河邊找鬼躬充。 笑死,一個(gè)胖子當(dāng)著我的面吹牛讨便,可吹牛的內(nèi)容都是我干的充甚。 我是一名探鬼主播,決...
    沈念sama閱讀 38,313評論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼器钟,長吁一口氣:“原來是場噩夢啊……” “哼津坑!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起傲霸,我...
    開封第一講書人閱讀 36,956評論 0 259
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后昙啄,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體穆役,經(jīng)...
    沈念sama閱讀 43,441評論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,925評論 2 323
  • 正文 我和宋清朗相戀三年梳凛,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了耿币。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,018評論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡韧拒,死狀恐怖淹接,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情叛溢,我是刑警寧澤塑悼,帶...
    沈念sama閱讀 33,685評論 4 322
  • 正文 年R本政府宣布,位于F島的核電站楷掉,受9級特大地震影響厢蒜,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜烹植,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,234評論 3 307
  • 文/蒙蒙 一斑鸦、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧草雕,春花似錦巷屿、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,240評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至败晴,卻和暖如春浓冒,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背尖坤。 一陣腳步聲響...
    開封第一講書人閱讀 31,464評論 1 261
  • 我被黑心中介騙來泰國打工稳懒, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人慢味。 一個(gè)月前我還...
    沈念sama閱讀 45,467評論 2 352
  • 正文 我出身青樓场梆,卻偏偏與公主長得像,于是被迫代替她去往敵國和親纯路。 傳聞我的和親對象是個(gè)殘疾皇子或油,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,762評論 2 345

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