问题:
给定一个二叉树
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
示例:
输入:root = [1,2,3,4,5,null,7]
输出:[1,#,2,3,#,4,5,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化输出按层序遍历顺序(由 next 指针连接),’#’ 表示每层的末尾。
思想:
基于树的层序遍历思想,首先建立一个链表用来存储每一层树的节点,然后循环遍历链表,直到为空,记录下每层节点的个数,然后循环,只要该该节点不是本身,他的next节点就是遍历到的节点,在此之前需要定义一个节点用来存储前一个节点。
代码:
class Solution {
public Node connect(Node root) {
if(root == null) return null;
Queue<Node> queue = new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
int size = queue.size();
Node n = null;
for(int i = 1;i<=size;i++){
Node r = queue.peek();
queue.poll();
if(r.left!=null){
queue.add(r.left);
}
if(r.right!=null){
queue.add(r.right);
}
if(i!=1){
n.next = r;
}
n = r;
}
}
return root;
}
}