【JS解决】leedcode 117. 填充每个节点的下一个右侧节点指针 II

题目要求：

给定一个二叉树

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。
初始状态下，所有 next 指针都被设置为 NULL。

示例：

输入：root = [1,2,3,4,5,null,7]
输出：[1,#,2,3,#,4,5,7,#]
解释：给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。序列化输出按层序遍历顺序（由 next 指针连接），‘#’ 表示每层的末尾。

JS代码

var connect = function(root) {
    
    // 处理特殊情况
    if (!root) return null
    // 构建层次遍历的的数组
    let arr = [root]
    while(arr.length){
    
        let n  = arr.length
        for(let i=0;i<n;i++){
    
            // 抛出当前节点
            let curNode = arr.shift()
            // 如果是最后一个节点则指向null
            if(i===n-1) curNode.next=null
            // 如果不是最后一个节点，则指向下个节点（也就是arr中的第一个节点）
            else curNode.next= arr[0]
            // 继续遍历左右孩子
            if(curNode.left!==null) arr.push(curNode.left)
            if(curNode.right!==null) arr.push(curNode.right)
        }
    }
    return root
};

运行结果