刻意練習:LeetCode實戰 -- 二叉樹的前序遍歷

背景

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


題目

  • 題號:144
  • 難度:中等
  • https://leetcode-cn.com/problems/binary-tree-preorder-traversal/

給定一個二叉樹,返回它的 前序 遍歷。

示例:

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

輸出: [1,2,3]

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


實現

第一種:利用棧

  • 執行結果:通過
  • 執行用時:276 ms, 在所有 C# 提交中擊敗了 84.15% 的用戶
  • 內存消耗:29.9 MB, 在所有 C# 提交中擊敗了 5.00% 的用戶
/**
 * 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> PreorderTraversal(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(lst.Count, root.val);
                root = root.left;
            }
            else
            {
                TreeNode node = stack.Pop();
                root = node.right;
            }
        }
        return lst;
    }
}

第二種:利用遞歸

  • 執行結果:通過
  • 執行用時:280 ms, 在所有 C# 提交中擊敗了 79.27% 的用戶
  • 內存消耗:29.8 MB, 在所有 C# 提交中擊敗了 5.00% 的用戶
/**
 * 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> PreorderTraversal(TreeNode root)
    {
        IList<int> lst = new List<int>();
        PreOrder(root, lst);
        return lst;
    }
    private void PreOrder(TreeNode node, IList<int> lst)
    {
        if (node == null)
            return;

        lst.Add(node.val);
        PreOrder(node.left, lst);
        PreOrder(node.right, lst);
    }
}

Python 語言

  • 執行結果:通過
  • 執行用時:36 ms, 在所有 Python3 提交中擊敗了 60.09% 的用戶
  • 內存消耗:13.7 MB, 在所有 Python3 提交中擊敗了 5.01% 的用戶
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

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

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

往期活動

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


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

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

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

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章