当前位置:   article > 正文

算法刷题打卡第21天:二叉树的前序遍历---迭代_root = [1,null,2,3]

root = [1,null,2,3]

二叉树的前序遍历

难度:简单
给你二叉树的根节点 root ,返回它节点值的前序遍历。

示例 1:

在这里插入图片描述

输入:root = [1,null,2,3]
输出:[1,2,3]
  • 1
  • 2

示例 2:

输入:root = []
输出:[]
  • 1
  • 2

示例 3:

输入:root = [1]
输出:[1]
  • 1
  • 2

示例 4:
在这里插入图片描述

输入:root = [1,2]
输出:[1,2]
  • 1
  • 2

示例 5:
在这里插入图片描述

输入:root = [1,null,2]
输出:[1,2]
  • 1
  • 2

迭代算法

思路:
首先我们需要了解什么是二叉树的前序遍历:按照访问根节点——左子树——右子树的方式遍历这棵树,而在访问左子树或者右子树的时候,我们按照同样的方式遍历,直到遍历完整棵树。因此整个遍历过程天然具有递归的性质,我们可以直接用递归函数来模拟这一过程。

定义 preorder(root) 表示当前遍历到 root 节点的答案。按照定义,我们只要首先将 root 节点的值加入答案,然后递归调用 preorder(root.left) 来遍历 root 节点的左子树,最后递归调用 preorder(root.right) 来遍历 root 节点的右子树即可,递归终止的条件为碰到空节点。

用迭代的方式实现上述的递归函数,两种方式是等价的,区别在于递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来,其余的实现与细节都相同,具体可以参考下面的代码。

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

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        results_lis = []
        if root is not None:
            node_lis = []
            node_lis.append(root)
            while len(node_lis) != 0:
                now_node = node_lis.pop()
                if now_node is not None:
                    results_lis.append(now_node.val)
                    node_lis.append(now_node.right)
                    node_lis.append(now_node.left)
        return results_lis
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/binary-tree-preorder-traversal

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/50444
推荐阅读
相关标签
  

闽ICP备14008679号