刻意練習：LeetCode實戰 -- 二叉樹的後序遍歷

背景

今天，第二期基礎算法（Leetcode）刻意練習訓練營 的打卡任務是二叉樹的中序遍歷，由於二叉樹的遍歷方式通常來說有四種：前序遍歷、中序遍歷、後序遍歷以及層次遍歷，而LeetCode也有二叉樹的後序遍歷題目，故一起寫了。

---

題目

• 題號：145
• 難度：困難
• https://leetcode-cn.com/problems/binary-tree-postorder-traversal/

給定一個二叉樹，返回它的 後序 遍歷。

示例:

輸入: [1,null,2,3]  
   1
    \
     2
    /
   3 

輸出: [3,2,1]

進階: 遞歸算法很簡單，你可以通過迭代算法完成嗎？

---

實現

第一種：利用棧

前序遍歷順序爲：根 -> 左 -> 右

後序遍歷順序爲：左 -> 右 -> 根

如果1：我們將前序遍歷中節點插入結果鏈表尾部的邏輯，修改爲將節點插入結果鏈表的頭部

那麼結果鏈表就變爲了：右 -> 左 -> 根

如果2：我們將遍歷的順序由從左到右修改爲從右到左，配合如果1

那麼結果鏈表就變爲了：左 -> 右 -> 根

這剛好是後序遍歷的順序

基於這兩個思路，我們處理方式如下：

第一：修改前序遍歷代碼中，節點寫入結果鏈表的代碼，將插入隊尾修改爲插入隊首

第二：修改前序遍歷代碼中，每次先查看左節點再查看右節點的邏輯，變爲先查看右節點再查看左節點

• 執行結果：通過
• 執行用時：272 ms, 在所有 C# 提交中擊敗了 98.15% 的用戶
• 內存消耗：30 MB, 在所有 C# 提交中擊敗了 6.90% 的用戶

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left;
 *     public TreeNode right;
 *     public TreeNode(int x) { val = x; }
 * }
 */
public class Solution
{
    public IList<int> PostorderTraversal(TreeNode root)
    {
        IList<int> lst = new List<int>();
        Stack<TreeNode> stack = new Stack<TreeNode>();
        while (stack.Count != 0 || root != null)
        {
            if (root != null)
            {
                stack.Push(root);
                lst.Insert(0, root.val);
                root = root.right;
            }
            else
            {
                TreeNode node = stack.Pop();
                root = node.left;
            }
        }
        return lst;
    }
}

第二種：利用遞歸

• 執行結果：通過
• 執行用時：276 ms, 在所有 C# 提交中擊敗了 89.81% 的用戶
• 內存消耗：30 MB, 在所有 C# 提交中擊敗了 6.90% 的用戶

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left;
 *     public TreeNode right;
 *     public TreeNode(int x) { val = x; }
 * }
 */
public class Solution
{
    public IList<int> PostorderTraversal(TreeNode root)
    {
        IList<int> lst = new List<int>();
        PostOrder(root, lst);
        return lst;
    }
    private void PostOrder(TreeNode node, IList<int> lst)
    {
        if (node == null)
            return;
            
        PostOrder(node.left, lst);
        PostOrder(node.right, lst);
        lst.Add(node.val);
    }
}

Python 語言

• 執行結果：通過
• 執行用時：28 ms, 在所有 Python3 提交中擊敗了 93.45% 的用戶
• 內存消耗：13.6 MB, 在所有 Python3 提交中擊敗了 5.09% 的用戶

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def postorderTraversal(self, root: TreeNode) -> List[int]:
        lst = list()
        self.postOrder(root, lst)
        return lst

    def postOrder(self, node: TreeNode, lst: list):
        if node is None:
            return
        self.postOrder(node.left, lst)
        self.postOrder(node.right, lst)
        lst.append(node.val)

---

往期活動

LSGO軟件技術團隊會定期開展提升編程技能的刻意練習活動，希望大家能夠參與進來一起刻意練習，一起學習進步！

• Python基礎刻意練習活動即將開啓，你參加嗎？
• Task01：變量、運算符與數據類型
• Task02：條件與循環
• Task03：列表與元組
• Task04：字符串與序列
• Task05：函數與Lambda表達式
• Task06：字典與集合
• Task07：文件與文件系統
• Task08：異常處理
• Task09：else 與 with 語句
• Task10：類與對象
• Task11：魔法方法
• Task12：模塊

---

我是 終身學習者“老馬”，一個長期踐行“結伴式學習”理念的 中年大叔。

我崇尚分享，渴望成長，於2010年創立了“LSGO軟件技術團隊”，並加入了國內著名的開源組織“Datawhale”，也是“Dre@mtech”、“智能機器人研究中心”和“大數據與哲學社會科學實驗室”的一員。

願我們一起學習，一起進步，相互陪伴，共同成長。

後臺回覆「搜搜搜」，隨機獲取電子資源！
歡迎關注，請掃描二維碼：