填充每个节点的下一个右侧节点指针 II
- 给定一个二叉树:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
-
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL 。
-
初始状态下,所有 next 指针都被设置为 NULL 。
示例 1:
输入:root = [1,2,3,4,5,null,7]
输出:[1,#,2,3,#,4,5,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化输出按层序遍历顺序(由 next 指针连接),‘#’ 表示每层的末尾。
解题思路
二叉树层序遍历中,把每层的元素通过next指针连接即可
- 1、使用队列进行层次遍历。初始时将根节点加入队列。
- 2、每次处理队列中的所有节点,设置当前层节点的 next 指针:
- 遍历当前层的节点时,将当前节点的 next 指针指向队列中的下一个节点(同一层的下一个节点)。
- 每处理完一个节点后,将其左右子节点加入队列,以便在下一层处理中使用。
- 3、继续上述过程直到队列为空。
Java实现
public class ConnectRightNode {
static class Node {
int val;
Node left;
Node right;
Node next;
Node(int x) {
val = x;
}
}
public Node connect(Node root) {
if (root == null) {
return null;
}
Queue<Node> queue =