[路飞]_leetcode-589-N 叉树的前序遍历

[题目地址]

给定一个 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] 内

递归解题

首先递归方法很简单，只需要处理当前节点，然后递归处理当前节点的子节点数组即可

代码如下：

var preorder = function(root) {
  // 创建结果数组
  const res = [];
  // 递归函数
  function _preorder(root){
    if(root === null) return;
    // 处理当前节点
    res.push(root.val);
    // 从左向右处理子节点
    for(let i = 0;i<root.children.length;i++){
      _preorder(root.children[i])
    }
  }
  _preorder(root);
  // 返回结果数组
  return res;
};
复制代码

迭代解题

这里的迭代算法遍历N叉树和之前的 leetcode-144-二叉树的前序遍历-迭代算法 是一样的思路

当前节点不为空的时候，一直让它等于左子树，过程中将未处理的节点用栈记录

当前节点为空时，处理栈顶元素

当前节点为空且栈为空，说明 N 叉树遍历完成，退出循环，返回结果数组即可

动画演示如下：

代码如下：

var preorder = function(root) {
  // 创建栈和结果数组
  const stack = [],res = [];
  // 当前节点不为空或者栈不为空的时候，遍历N叉树
  while(root||stack.length){
    // 节点不为空的时候
    while(root){
      // 当前节点值放入结果数组
      res.push(root.val);
      // 从右向左将子树入栈
      for(let i = root.children.length-1;i>0;i--){
        stack.push(root.children[i])
      }
      // 最左侧子树直接处理
      root = root.children[0];
    }
    // 当节点为空时，处理栈顶节点
    root = stack.pop();
  }
  // 返回结果数组
  return res;
}
复制代码

至此我们就完成了 leetcode-589-N 叉树的前序遍历

如有任何问题或建议，欢迎留言讨论！