Lstmxx的空间

117. 填充每个节点的下一个右侧节点指针 II

给定一个二叉树

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

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

初始状态下，所有 next 指针都被设置为 NULL。

链接

var connect = function(root) {
  if (root === null) return null
  const nodeList = [root]
  while(nodeList.length > 0) {
    const nodeL = nodeList.splice(0)
    const len = nodeL.length
    for(let i = 0; i < len; i++) {
      const node = nodeL[i]
      if (node === null) {
        continue
      }
      const back = nodeL[i + 1]
      node.next = back ? back : null
      node.left && nodeList.push(node.left)
      node.right && nodeList.push(node.right)
    }
  }
  return root
}