leetcode-589 N 叉树的前序遍历
b站视频

题目介绍

给定一个 N 叉树，返回其节点值的 前序遍历 。

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

进阶：

递归法很简单，你可以使用迭代法完成此题吗?

示例1

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

示例2

输入：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]

提示：

• N 叉树的高度小于或等于 1000
• 节点总数在范围 [0, 10^4] 内

解题思路

递归

N 叉树的前序遍历与二叉树的前序遍历差不多，先遍历树的根节点，然后从左到右依次遍历根节点的所有子树

解题代码

var preorder = function(root) {
    const ans = []
    return _preorder(root, ans)
};

var _preorder = function(root, ans) {
    if (!root) return ans
    // 将根节点的值 push 到结果数组中
    ans.push(root.val)
    // 从左到右依次进行子树的前序遍历
    for (const child of root.children) {
        _preorder(child, ans)
    }
    // 返回结果数组
    return ans
}

迭代

迭代法主要是利用栈的思想，步骤如下

1. 维护两个数组，一个是最后的结果数组 ans， 一个是存放节点的数组 treeNodeArr，一开始先将根节点放到 treeNodeArr 中
2. 当 treeNodeArr 数组不为空时，每次将 treeNodeArr 的第一个节点从数组中弹出，然后将该节点的值 push 到结果数组中
3. 如果当前节点有 children，则将当前节点的子节点从后往前依次从 treeNode 数组的前面插入
4. 重复 2-3 的步骤，直到 treeNodeArr 数组为空，返回 ans 数组即为最终结果

解题代码

var preorder = function(root) {
    // 如果是空树，直接返回 []
    if (!root) return []
    // 定义 res 数组用于存放最后的遍历结果
    const res = []
    // 定义 treeNodeArr 数组用于存放树的节点
    const treeNodeArr = [root]
    // 如果 treeNodeArr 不为空，说明还没有遍历完整棵树
    while (treeNodeArr.length) {
        // 将当前的根节点（树或子树）从 treeNodeArr 数组中弹出
        const node = treeNodeArr.shift()
        // 从后往前遍历根节点的子节点，并从 treeNodeArr 的前面依次插入
        for (let i = node.children.length; i > 0; i--) {
            treeNodeArr.unshift(node.children[i - 1])
        }
        // 将根节点的值 push 到结果数组中
        res.push(node.val)
    }
    // 返回最终结果
    return res
};