Construct Binary Tree from Preorder and Inorder Traversal

https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/description/

public TreeNode buildTree(int[] preorder, int[] inorder) {
    Map<Integer, Integer> inMap = new HashMap<Integer, Integer>();

    for(int i = 0; i < inorder.length; i++) {
        inMap.put(inorder[i], i);
    }

    TreeNode root = buildTree(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1, inMap);
    return root;
}

public TreeNode buildTree(int[] preorder, int preStart, int preEnd, int[] inorder, int inStart, int inEnd, Map<Integer, Integer> inMap) {
    if(preStart > preEnd || inStart > inEnd) return null;

    TreeNode root = new TreeNode(preorder[preStart]);
    int inRoot = inMap.get(root.val);
    int numsLeft = inRoot - inStart;//root的左子節(jié)點(diǎn)的數(shù)量

    root.left = buildTree(preorder, preStart + 1, preStart + numsLeft, inorder, inStart, inRoot - 1, inMap);
//在前序遍歷對(duì)應(yīng)的vector中牲尺,如果root有左子節(jié)點(diǎn)的話那么對(duì)應(yīng)的為vector中它對(duì)應(yīng)的下一個(gè)節(jié)點(diǎn)
    root.right = buildTree(preorder, preStart + numsLeft + 1, preEnd, inorder, inRoot + 1, inEnd, inMap);
//在前序遍歷對(duì)應(yīng)的vector中伴挚,如果root有右節(jié)點(diǎn)的話遮婶,那么對(duì)應(yīng)的右節(jié)點(diǎn)在vector中的索引為preStart + numsLeft + 1(numsLeft 是左節(jié)點(diǎn)的數(shù)量)
    return root;
}

另外一種不使用遞歸的算法

class Solution {                                                        
public:
    TreeNode* buildTree(vector<int> &pre, vector<int> &in) {           
           int i = 0, j = 0;
           TreeNode *root = new TreeNode(0x80000000);
           TreeNode *pp = NULL,*ptr = root;
           stack<TreeNode*> sn;
           sn.push(root);
        
           while (j<in.size()) {
            if(sn.top()->val == in[j]){//當(dāng)該節(jié)點(diǎn)與 中序遍歷的j節(jié)點(diǎn)相同說(shuō)明 當(dāng)前前序遍歷數(shù)組中的i(即當(dāng)前棧頂節(jié)點(diǎn)對(duì)應(yīng)的下一個(gè)節(jié)點(diǎn))所對(duì)應(yīng)的節(jié)點(diǎn)為sn棧頂節(jié)點(diǎn)的右節(jié)點(diǎn)
                                       //即相當(dāng)于當(dāng)前的棧頂節(jié)點(diǎn)已經(jīng)”鏈接“上了左節(jié)點(diǎn)那么下一個(gè)索引i對(duì)應(yīng)的節(jié)點(diǎn)應(yīng)為該節(jié)點(diǎn)的右節(jié)點(diǎn)                      
               pp = sn.top();
                sn.pop();
                j++;
            }else if(pp){
                ptr = new TreeNode(pre[i]);
                pp->right = ptr;
                pp = NULL;
                sn.push(ptr);
                i++;
            }else{//一直左左左
                ptr = new TreeNode(pre[i]);
                sn.top()->left = ptr;
                sn.push(ptr);
                i++;
            }
          }      
          ptr = root->left;
          delete root;
          return ptr;
    }
};
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子芭碍,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,372評(píng)論 6 498
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件姐浮,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡葬馋,警方通過(guò)查閱死者的電腦和手機(jī)卖鲤,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,368評(píng)論 3 392
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)畴嘶,“玉大人蛋逾,你說(shuō)我怎么就攤上這事〈懊酰” “怎么了区匣?”我有些...
    開(kāi)封第一講書(shū)人閱讀 162,415評(píng)論 0 353
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)蒋院。 經(jīng)常有香客問(wèn)我亏钩,道長(zhǎng),這世上最難降的妖魔是什么悦污? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 58,157評(píng)論 1 292
  • 正文 為了忘掉前任铸屉,我火速辦了婚禮,結(jié)果婚禮上切端,老公的妹妹穿的比我還像新娘彻坛。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,171評(píng)論 6 388
  • 文/花漫 我一把揭開(kāi)白布昌屉。 她就那樣靜靜地躺著钙蒙,像睡著了一般。 火紅的嫁衣襯著肌膚如雪间驮。 梳的紋絲不亂的頭發(fā)上躬厌,一...
    開(kāi)封第一講書(shū)人閱讀 51,125評(píng)論 1 297
  • 那天,我揣著相機(jī)與錄音竞帽,去河邊找鬼扛施。 笑死,一個(gè)胖子當(dāng)著我的面吹牛屹篓,可吹牛的內(nèi)容都是我干的疙渣。 我是一名探鬼主播,決...
    沈念sama閱讀 40,028評(píng)論 3 417
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼堆巧,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼妄荔!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起谍肤,我...
    開(kāi)封第一講書(shū)人閱讀 38,887評(píng)論 0 274
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤啦租,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后荒揣,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體篷角,經(jīng)...
    沈念sama閱讀 45,310評(píng)論 1 310
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,533評(píng)論 2 332
  • 正文 我和宋清朗相戀三年系任,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了内地。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,690評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡赋除,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出非凌,到底是詐尸還是另有隱情举农,我是刑警寧澤,帶...
    沈念sama閱讀 35,411評(píng)論 5 343
  • 正文 年R本政府宣布敞嗡,位于F島的核電站颁糟,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏喉悴。R本人自食惡果不足惜棱貌,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,004評(píng)論 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望箕肃。 院中可真熱鬧婚脱,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 31,659評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至篮洁,卻和暖如春涩维,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背袁波。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 32,812評(píng)論 1 268
  • 我被黑心中介騙來(lái)泰國(guó)打工瓦阐, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人篷牌。 一個(gè)月前我還...
    沈念sama閱讀 47,693評(píng)論 2 368
  • 正文 我出身青樓睡蟋,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親娃磺。 傳聞我的和親對(duì)象是個(gè)殘疾皇子薄湿,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,577評(píng)論 2 353

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