二叉樹遍歷(先序、中序容诬、后序)

二叉樹遍歷 by Java

二叉樹有多種遍歷方法娩梨,有層次遍歷、深度優(yōu)先遍歷览徒、廣度優(yōu)先遍歷等狈定。

本文只涉及二叉樹的先序、中序习蓬、后序的遞歸和非遞歸遍歷纽什。

涉及到的代碼都用Java編寫。


首先給出二叉樹節(jié)點(diǎn)類:

樹節(jié)點(diǎn):

class TreeNode {
    int val;
    //左子樹
    TreeNode left;
    //右子樹
    TreeNode right;
    //構(gòu)造方法
    TreeNode(int x) {
        val = x;
    }
}

無論是哪種遍歷方法躲叼,考查節(jié)點(diǎn)的順序都是一樣的(思考做試卷的時候芦缰,人工遍歷考查順序)。只不過有時候考查了節(jié)點(diǎn)枫慷,將其暫存让蕾,需要之后的過程中輸出浪规。

圖2:先序、中序探孝、后序遍歷節(jié)點(diǎn)考查順序

如圖1所示笋婿,三種遍歷方法(人工)得到的結(jié)果分別是:

先序:1 2 4 6 7 8 3 5
中序:4 7 6 8 2 1 3 5
后序:7 8 6 4 2 5 3 1

三種遍歷方法的考查順序一致,得到的結(jié)果卻不一樣顿颅,原因在于:

先序:考察到一個節(jié)點(diǎn)后缸濒,即刻輸出該節(jié)點(diǎn)的值,并繼續(xù)遍歷其左右子樹粱腻。(根左右)

中序:考察到一個節(jié)點(diǎn)后庇配,將其暫存,遍歷完左子樹后栖疑,再輸出該節(jié)點(diǎn)的值讨永,然后遍歷右子樹滔驶。(左根右)

后序:考察到一個節(jié)點(diǎn)后遇革,將其暫存,遍歷完左右子樹后揭糕,再輸出該節(jié)點(diǎn)的值萝快。(左右根)


先序遍歷

遞歸先序遍歷

遞歸先序遍歷很容易理解,先輸出節(jié)點(diǎn)的值著角,再遞歸遍歷左右子樹揪漩。中序和后序的遞歸類似,改變根節(jié)點(diǎn)輸出位置即可吏口。

// 遞歸先序遍歷
public static void recursionPreorderTraversal(TreeNode root) {
    if (root != null) {
        System.out.print(root.val + " ");
        recursionPreorderTraversal(root.left);
        recursionPreorderTraversal(root.right);
    }
}

非遞歸先序遍歷

因?yàn)橐诒闅v完節(jié)點(diǎn)的左子樹后接著遍歷節(jié)點(diǎn)的右子樹奄容,為了能找到該節(jié)點(diǎn),需要使用來進(jìn)行暫存产徊。中序和后序也都涉及到回溯昂勒,所以都需要用到

圖2:非遞歸先序遍歷

遍歷過程參考注釋

// 非遞歸先序遍歷
public static void preorderTraversal(TreeNode root) {
    // 用來暫存節(jié)點(diǎn)的棧
    Stack<TreeNode> treeNodeStack = new Stack<TreeNode>();
    // 新建一個游標(biāo)節(jié)點(diǎn)為根節(jié)點(diǎn)
    TreeNode node = root;
    // 當(dāng)遍歷到最后一個節(jié)點(diǎn)的時候舟铜,無論它的左右子樹都為空戈盈,并且棧也為空
    // 所以,只要不同時滿足這兩點(diǎn)谆刨,都需要進(jìn)入循環(huán)
    while (node != null || !treeNodeStack.isEmpty()) {
        // 若當(dāng)前考查節(jié)點(diǎn)非空塘娶,則輸出該節(jié)點(diǎn)的值
        // 由考查順序得知,需要一直往左走
        while (node != null) {
            System.out.print(node.val + " ");
            // 為了之后能找到該節(jié)點(diǎn)的右子樹痊夭,暫存該節(jié)點(diǎn)
            treeNodeStack.push(node);
            node = node.left;
        }
        // 一直到左子樹為空刁岸,則開始考慮右子樹
        // 如果棧已空,就不需要再考慮
        // 彈出棧頂元素她我,將游標(biāo)等于該節(jié)點(diǎn)的右子樹
        if (!treeNodeStack.isEmpty()) {
            node = treeNodeStack.pop();
            node = node.right;
        }
    }
}

先序遍歷結(jié)果:

遞歸先序遍歷: 1 2 4 6 7 8 3 5
非遞歸先序遍歷:1 2 4 6 7 8 3 5


中序遍歷

遞歸中序遍歷

過程和遞歸先序遍歷類似

// 遞歸中序遍歷
public static void recursionMiddleorderTraversal(TreeNode root) {
    if (root != null) {
        recursionMiddleorderTraversal(root.left);
        System.out.print(root.val + " ");
        recursionMiddleorderTraversal(root.right);
    }
}

非遞歸中序遍歷

和非遞歸先序遍歷類似虹曙,唯一區(qū)別是考查到當(dāng)前節(jié)點(diǎn)時膝宁,并不直接輸出該節(jié)點(diǎn)。

而是當(dāng)考查節(jié)點(diǎn)為空時根吁,從棧中彈出的時候再進(jìn)行輸出(永遠(yuǎn)先考慮左子樹员淫,直到左子樹為空才訪問根節(jié)點(diǎn))。

// 非遞歸中序遍歷
public static void middleorderTraversal(TreeNode root) {
    Stack<TreeNode> treeNodeStack = new Stack<TreeNode>();
    TreeNode node = root;
    while (node != null || !treeNodeStack.isEmpty()) {
        while (node != null) {
            treeNodeStack.push(node);
            node = node.left;
        }
        if (!treeNodeStack.isEmpty()) {
            node = treeNodeStack.pop();
            System.out.print(node.val + " ");
            node = node.right;
        }
    }
}

中序遍歷結(jié)果

遞歸中序遍歷: 4 7 6 8 2 1 3 5
非遞歸中序遍歷:4 7 6 8 2 1 3 5


后序遍歷

遞歸后序遍歷

過程和遞歸先序遍歷類似

// 遞歸后序遍歷
public static void recursionPostorderTraversal(TreeNode root) {
    if (root != null) {
        recursionPostorderTraversal(root.left);
        recursionPostorderTraversal(root.right);
        System.out.print(root.val + " ");
    }
}

非遞歸后序遍歷

后續(xù)遍歷和先序击敌、中序遍歷不太一樣介返。

后序遍歷在決定是否可以輸出當(dāng)前節(jié)點(diǎn)的值的時候,需要考慮其左右子樹是否都已經(jīng)遍歷完成沃斤。

所以需要設(shè)置一個lastVisit游標(biāo)圣蝎。

若lastVisit等于當(dāng)前考查節(jié)點(diǎn)的右子樹,表示該節(jié)點(diǎn)的左右子樹都已經(jīng)遍歷完成衡瓶,則可以輸出當(dāng)前節(jié)點(diǎn)徘公。

并把lastVisit節(jié)點(diǎn)設(shè)置成當(dāng)前節(jié)點(diǎn),將當(dāng)前游標(biāo)節(jié)點(diǎn)node設(shè)置為空哮针,下一輪就可以訪問棧頂元素关面。

否者,需要接著考慮右子樹十厢,node = node.right等太。

以下考慮后序遍歷中的三種情況:

圖3:后序,右子樹不為空蛮放,node = node.right

如圖3所示缩抡,從節(jié)點(diǎn)1開始考查直到節(jié)點(diǎn)4的左子樹為空。

注:此時的游標(biāo)節(jié)點(diǎn)node = 4.left == null包颁。

此時需要從棧中查看 Peek()棧頂元素瞻想。

發(fā)現(xiàn)節(jié)點(diǎn)4的右子樹非空,需要接著考查右子樹娩嚼,4不能輸出蘑险,node = node.right。

圖4:后序待锈,左右子樹都為空漠其,直接輸出

如圖4所示,考查到節(jié)點(diǎn)7(7.left == null竿音,7是從棧中彈出)和屎,其左右子樹都為空,可以直接輸出7春瞬。

此時需要把lastVisit設(shè)置成節(jié)點(diǎn)7柴信,并把游標(biāo)節(jié)點(diǎn)node設(shè)置成null,下一輪循環(huán)的時候會考查棧中的節(jié)點(diǎn)6宽气。

圖5:后序随常,右子樹 = lastVisit潜沦,直接輸出

如圖5所示,考查完節(jié)點(diǎn)8之后(lastVisit == 節(jié)點(diǎn)8)绪氛,將游標(biāo)節(jié)點(diǎn)node賦值為棧頂元素6唆鸡,節(jié)點(diǎn)6的右子樹正好等于節(jié)點(diǎn)8。表示節(jié)點(diǎn)6的左右子樹都已經(jīng)遍歷完成枣察,直接輸出6争占。

此時,可以將節(jié)點(diǎn)直接從棧中彈出Pop()序目,之前用的只是Peek()臂痕。

將游標(biāo)節(jié)點(diǎn)node設(shè)置成null。

// 非遞歸后序遍歷
public static void postorderTraversal(TreeNode root) {
    Stack<TreeNode> treeNodeStack = new Stack<TreeNode>();
    TreeNode node = root;
    TreeNode lastVisit = root;
    while (node != null || !treeNodeStack.isEmpty()) {
        while (node != null) {
            treeNodeStack.push(node);
            node = node.left;
        }
        //查看當(dāng)前棧頂元素
        node = treeNodeStack.peek();
        //如果其右子樹也為空猿涨,或者右子樹已經(jīng)訪問
        //則可以直接輸出當(dāng)前節(jié)點(diǎn)的值
        if (node.right == null || node.right == lastVisit) {
            System.out.print(node.val + " ");
            treeNodeStack.pop();
            lastVisit = node;
            node = null;
        } else {
            //否則握童,繼續(xù)遍歷右子樹
            node = node.right;
        }
    }
}

后序遍歷結(jié)果

遞歸后序遍歷: 7 8 6 4 2 5 3 1
非遞歸后序遍歷:7 8 6 4 2 5 3 1


有任何錯誤請指正,謝謝叛赚!
寫文字不容易澡绩,且寫且珍惜!

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末红伦,一起剝皮案震驚了整個濱河市英古,隨后出現(xiàn)的幾起案子淀衣,更是在濱河造成了極大的恐慌昙读,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,324評論 6 498
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件膨桥,死亡現(xiàn)場離奇詭異蛮浑,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)只嚣,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,356評論 3 392
  • 文/潘曉璐 我一進(jìn)店門沮稚,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人册舞,你說我怎么就攤上這事蕴掏。” “怎么了调鲸?”我有些...
    開封第一講書人閱讀 162,328評論 0 353
  • 文/不壞的土叔 我叫張陵盛杰,是天一觀的道長。 經(jīng)常有香客問我藐石,道長即供,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,147評論 1 292
  • 正文 為了忘掉前任于微,我火速辦了婚禮逗嫡,結(jié)果婚禮上青自,老公的妹妹穿的比我還像新娘。我一直安慰自己驱证,他們只是感情好延窜,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,160評論 6 388
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著抹锄,像睡著了一般需曾。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上祈远,一...
    開封第一講書人閱讀 51,115評論 1 296
  • 那天呆万,我揣著相機(jī)與錄音,去河邊找鬼车份。 笑死谋减,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的扫沼。 我是一名探鬼主播出爹,決...
    沈念sama閱讀 40,025評論 3 417
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼缎除!你這毒婦竟也來了严就?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,867評論 0 274
  • 序言:老撾萬榮一對情侶失蹤器罐,失蹤者是張志新(化名)和其女友劉穎梢为,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體轰坊,經(jīng)...
    沈念sama閱讀 45,307評論 1 310
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡铸董,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,528評論 2 332
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了肴沫。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片粟害。...
    茶點(diǎn)故事閱讀 39,688評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖颤芬,靈堂內(nèi)的尸體忽然破棺而出悲幅,到底是詐尸還是另有隱情,我是刑警寧澤站蝠,帶...
    沈念sama閱讀 35,409評論 5 343
  • 正文 年R本政府宣布汰具,位于F島的核電站,受9級特大地震影響沉衣,放射性物質(zhì)發(fā)生泄漏郁副。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,001評論 3 325
  • 文/蒙蒙 一豌习、第九天 我趴在偏房一處隱蔽的房頂上張望存谎。 院中可真熱鬧拔疚,春花似錦、人聲如沸既荚。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,657評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽恰聘。三九已至句各,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間晴叨,已是汗流浹背凿宾。 一陣腳步聲響...
    開封第一講書人閱讀 32,811評論 1 268
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留兼蕊,地道東北人初厚。 一個月前我還...
    沈念sama閱讀 47,685評論 2 368
  • 正文 我出身青樓,卻偏偏與公主長得像孙技,于是被迫代替她去往敵國和親产禾。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,573評論 2 353

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