题目:
给定一个二叉树:
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;
}