Leetcode练习:144. 二叉树的前序遍历(含遍历方式)

问题描述

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

示例 1: 输入:root = [1,null,2,3] 输出:[1,2,3]
示例 2: 输入:root = [] 输出:[]
示例 3: 输入:root = [1] 输出:[1] 示例 4: 输入:root = [1,2] 输出:[1,2]
示例 5: 输入:root = [1,null,2] 输出:[1,2]

提示:

树中节点数目在范围 [0, 100] 内 -100 <= Node.val <= 100

思路分析及代码实现

前序遍历:

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

迭代: 利用栈来进行class Solution: 先入栈右孩子,再入栈左孩子,这样在出栈时就先出左孩子,也是按照根节点–左孩子–右孩子的顺序

def preorderTraversal(self, root: TreeNode) -> List[int]:
    if not root:
        return []
    stack = [root]
    res = []
    while stack:
        node = stack.pop()
        if node:
            res.append(node.val)
            if node.right:  # 先将右孩子入栈
                stack.append(node.right)
            if node.left:  # 再将左孩子入栈
                stack.append(node.left)
    return res

遍历方式

如果按照 根节点 -> 左孩子 -> 右孩子 的方式遍历,即「前序遍历」,每次先遍历根节点,遍历结果为 1 2 4 5 3 6 7;

同理,如果按照 左孩子 -> 根节点 -> 右孩子 的方式遍历,即「中序序遍历」,遍历结果为 4 2 5 1 6 3 7;

如果按照 左孩子 -> 右孩子 -> 根节点 的方式遍历,即「后序序遍历」,遍历结果为 4 5 2 6 7 3 1;

最后,层次遍历就是按照每一层从左向右的方式进行遍历,遍历结果为 1 2 3 4 5 6 7。

经验分享 程序员 微信小程序 职场和发展