给定一个 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 叉树的前序遍历
如有任何问题或建议,欢迎留言讨论!