给定一个 完美二叉树 ，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

1
2
3
4
5
6
struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

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

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

进阶：

  • 你只能使用常量级额外空间。
  • 使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

思路

  1. 和 《116 - 填充每个节点的下一个右侧节点指针》 没有区别，只是限定使用常数级额外空间
  2. 层次遍历进而填充 next
  3. 使用一个 dummyHead 节点作为链表头维护
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
var connect = function (root) {
  let dummyHead = new Node()
  dummyHead.next = root
  while (dummyHead.next) {
    let cur = dummyHead.next
    let nextHead = new Node()
    let nextCur = nextHead
    while (cur) {
      if (cur.left) {
        nextCur.next = cur.left
        nextCur = nextCur.next
      }
      if (cur.right) {
        nextCur.next = cur.right
        nextCur = nextCur.next
      }
      cur = cur.next
    }
    dummyHead = nextHead
  }
  return root
}