LeetCode 589. N 叉树的前序遍历

30 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第18天,点击查看活动详情

一、题目描述:

589. N 叉树的前序遍历

给定一个 n 叉树的根节点  root ,返回 其节点值的 前序遍历 。

n 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。

示例 1:

image.png

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

示例 2:

image.png

输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[1,2,3,6,7,11,14,4,8,12,5,9,13,10]

提示:

  • 节点总数在范围 [0, 104]内
  • 0 <= Node.val <= 104
  • n 叉树的高度小于或等于 1000  

进阶:递归法很简单,你可以使用迭代法完成此题吗?

二、思路分析:

用栈辅助。

首先,将 root 压入栈中
在栈不为空时,对栈进行遍历,每次弹出栈顶元素
若栈顶元素节点不为空,则将该节点值放入结果集中,且将该节点的子节点从右至左压入栈中(这样弹出时就是从左至右,符合前序遍历的顺序)

三、AC 代码:

"""
# Definition for a Node.
class Node:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children
"""
class Solution:
    def preorder(self, root: 'Node') -> List[int]:
        stack = []
        stack.append(root)
        res = []
        while len(stack) > 0:
            top = stack.pop()
            if top is None:
                continue
            res.append(top.val)
            # 反序插入子节点
            children = top.children
            for i in range(len(children) - 1, -1, -1):
                child = children[i]
                stack.append(child)
        return res

范文参考:

N叉树的先序遍历在Trie树中应用广泛,用一个成员变量来收取每一个节点。 - N 叉树的前序遍历 - 力扣(LeetCode)

递归(JS引擎维护访问栈)、迭代(自己通过额外的数组维护访问栈) - N 叉树的前序遍历 - 力扣(LeetCode)