JiatLn/JiatLn.me

test a issues

Opened this issue · 0 comments

【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;
};