LeetCode144.二叉树的前序遍历（python）

Posted 2022-12-03 归止于飞

LeetCode144.前序遍历

题干

解法1

思路：递归
创建一个递归函数，注意递归的三要素：停止递归的条件，简化问题，父问题与子问题没有重叠

class Solution:
    def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        res = []
        def fun_pre(root):
            if root == None:
                return
            res.append(root.val)
            fun_pre(root.left)
            fun_pre(root.right)
        fun_pre(root)
        return res

时间复杂度分析：

时间复杂度：O(n)，其中 n 是二叉树的节点数。每一个节点恰好被遍历一次。
空间复杂度：O(n)，为递归过程中栈的开销，平均情况下为 O(logn)，最坏情况下树呈现链状，为O(n)。

解法2：

思路：迭代
其实递归解法与迭代解法几乎一样，递归是存在一个隐式栈，而迭代需要把它显式模拟出来。这也就是递归的本质，栈帧。

class Solution:
    def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        res = []
        s1 = []
        while s1 or root:
            if root != None:
                res.append(root.val)
                s1.append(root.right)
                root = root.left
            else:
                tmp = s1.pop()
                root = tmp
        return res

时间复杂度分析：

时间复杂度：O(n)，其中 n 是二叉树的节点数。每一个节点恰好被遍历一次。
空间复杂度：O(n)，为迭代过程中显式栈的开销，平均情况下为 O(logn)，最坏情况下树呈现链状，为 O(n)。