题目:
给定一个二叉树:
struct Node {int val;Node *left;Node *right;Node *next; }
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL
。
初始状态下,所有 next 指针都被设置为 NULL
。
可以使用层序遍历来解决这个问题。基本思路是:
- 使用队列进行层序遍历
- 对于每一层,将该层的节点连接起来
- 最后一个节点的next保持为null
- 首先,检查root是否为null。如果是,直接返回null。
- 创建一个队列来进行层序遍历。
- 使用一个while循环来遍历每一层。
- 对于每一层,先获取该层的节点数量(levelSize)。
- 然后,遍历该层的每个节点:
- 将节点从队列中取出
- 如果不是该层的最后一个节点,就将其next指向队列的下一个节点
- 如果该节点有左子节点,将左子节点加入队列
- 如果该节点有右子节点,将右子节点加入队列
- 重复这个过程,直到队列为空。
- 最后,返回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;
}