网络异常,图片无法展示
|
给定一个 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 叉树的前序遍历
如有任何问题或建议,欢迎留言讨论!