[leetcode hot 150]第一百一十七题，填充每个节点的下一个右侧节点

题目：
 

给定一个二叉树：

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

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

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

 

可以使用层序遍历来解决这个问题。基本思路是：

1. 使用队列进行层序遍历
2. 对于每一层，将该层的节点连接起来
3. 最后一个节点的next保持为null

1. 首先，检查root是否为null。如果是，直接返回null。
2. 创建一个队列来进行层序遍历。
3. 使用一个while循环来遍历每一层。
4. 对于每一层，先获取该层的节点数量（levelSize）。
5. 然后，遍历该层的每个节点：
   • 将节点从队列中取出
   • 如果不是该层的最后一个节点，就将其next指向队列的下一个节点
   • 如果该节点有左子节点，将左子节点加入队列
   • 如果该节点有右子节点，将右子节点加入队列
6. 重复这个过程，直到队列为空。
7. 最后，返回root节点。

public static TreeNode connect(TreeNode root) {
        if (root == null) return null;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int levelSize = queue.size();
            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll();
                if (i < levelSize - 1) {
                    node.next = queue.peek();
                }
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
        }

        return root;
}