【LeetCode】429. N 叉树的层序遍历

【LeetCode】429. N 叉树的层序遍历

给定一个 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;
};