填充每个节点的下一个右侧节点指针 II
中等
给定一个二叉树:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL 。
初始状态下,所有 next 指针都被设置为 NULL 。
示例 1:
![](https://img-blog.csdnimg.cn/img_convert/cd254bc286355f79c0b7f64e98153f8d.png)
输入:root = [1,2,3,4,5,null,7]
输出:[1,#,2,3,#,4,5,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化输出按层序遍历顺序(由 next 指针连接),'#' 表示每层的末尾。
示例 2:
输入:root = []
输出:[]
题解
先判空,处理节点为空的情况
利用队列的特点对节点进行层序遍历
while循环当队列元素 > 0说明队列还有节点,用一个变量接收此时队列元素大小,便于后面取出每一层的每一个节点
先从队列取出一个节点作为第一个节点,把该节点的左右子树进行入队列操作,在for循环遍历从1开始遍历取出队列节点,如果能取出也进行左右子树入队列操作,就把该节点作为第一个节点下一个节点,第一个节点后移,继续该操作,如果没有则不进行next操作,默认为null。
/*
// Definition for a Node.
class Node {
public int val;
public Node left;
public Node right;
public Node next;
public Node() {}
public Node(int _val) {
val = _val;
}
public Node(int _val, Node _left, Node _right, Node _next) {
val = _val;
left = _left;
right = _right;
next = _next;
}
};
*/
class Solution {
public Node connect(Node root) {
if(root == null){
return null;
}
Queue<Node> queue = new LinkedList<>();
queue.add(root);
while (queue.size() > 0){
int num = queue.size();
Node node6 = queue.remove();
if(node6.left != null){
queue.add(node6.left);
}
if(node6.right != null){
queue.add(node6.right);
}
for (int i = 1; i < num;i++) {
Node node7 = queue.remove();
if(node7.left != null){
queue.add(node7.left);
}
if(node7.right != null){
queue.add(node7.right);
}
node6.next = node7;
node6 = node6.next;
}
}
return root;
}
}