test a issues
Opened this issue · 0 comments
JiatLn commented
给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。
树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。
输入:
root = [1,null,3,2,4,null,5,6]
输出:[[1],[3,2,4],[5,6]]
var levelOrder = function(root) {
if(!root) return [];
let ans = [];
let stack = [[root]];
while(stack.length) {
let curr = stack.pop(); // 当前层的节点数组
let arr = []; // 记录当前层各节点的值
let level = []; // 保存当前层的子节点,用于下次遍历
// 遍历当前层,取出值,顺便把子节点放到level中
curr.forEach(node => {
arr.push(node.val); // 取出节点的值
node.children.forEach(ch => level.push(ch));
})
ans.push(arr); // 把当前层的值放到结果中
// 如果有子节点,则入栈,进行下一层遍历,否则跳出循环
level.length && stack.push(level);
}
return ans;
};