每日一题：填充每个节点的下一个右侧节点指针

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第3天，点击查看活动详情

题目链接

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

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

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

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

示例 1：

输入： root = [1,2,3,4,5,6,7]
输出： [1,#,2,3,#,4,5,6,7,#]
解释： 给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。序列化的输出按层序遍历排列，同一层节点由 next 指针连接，'#' 标志着每一层的结束。

示例 2:

输入： root = []
输出： []

提示：

• 树中节点的数量在 [0, 212 - 1] 范围内
• -1000 <= node.val <= 1000

解题思路

本题可以使用层序遍历，在遍历每层的数据时，记录一下每层的头节点，然后再遍历后面的数据时，将前一个节点的指针指向本节点就可以

具体步骤：

1. 在每层遍历时，记录这一层的头节点
2. 在下一次遍历时，将找到的头节点的next指向本节点，并更新头节点的位置
3. 一层遍历完成后，将本轮的最后一个节点next指向null

具体代码如下：

 public Node connect(Node root) {
        Queue<Node> queue = new LinkedList<>();
        if (root != null) {
            queue.offer(root);
        }

        while (!queue.isEmpty()){
            int size = queue.size();
            Node slowNode = null;
            Node node = null;
            for (int i = 0; i < size;i++) {
                if (i == 0) {
                    slowNode = queue.poll();//取出头节点
                    node = slowNode;
                }else {
                    node = queue.poll();
                    slowNode.next = node;//将前一个节点next指向本节点   
                    slowNode = slowNode.next;
                }

                if (node.left != null) {
                    queue.offer(node.left);
                }

                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            slowNode.next = null;//将点最后一个next指向null
        }
        return root;
    }

其他

力扣上还有一道与之类似的题填充每个节点的下一个右侧节点指针 II，区别就在于这道题说的是二叉树，不一定是完全二叉树，其实没有什么差别，思路和逻辑和上面的其实都一样