一、题目内容
二、题目分析
此题我们采用广度优先遍历算法进行解题,我们发现,每个节点的next总是指向同层的右边的那个节点(最后一个节点指向null)。那么我们是不是可以将节点按层一层层的放入队列,然后让他们的next依次等于下一个节点即可呢?
根据BFS基本套路,首先我们要while循环,里面的条件是队列不可以为空,如果就这样不加限制的话,最终的结果是每个节点都指向它先序遍历的下个节点,那是因为当前这层的最后一个节点会指向下一层的第一个节点。所以我们正确的做法是,在while循环中加一个对当前队列长度的变量,再根据这个变量建立另外一个循环,将当前队列所含有的节点的子节点都添加进去,然后设置一个额外的空间来交替更新next的值即可得出结果、
代码:
public Node connect(Node root) {
if(root==null)
return root;
Queue<Node>queue=new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
int size=queue.size();
Node last=null;
for(int i=0;i<size;i++){
Node tmp=queue.remove();
if(tmp.left!=null)
queue.add(tmp.left);
if(tmp.right!=null)
queue.add(tmp.right);
if(i!=0)
last.next=tmp;
last=tmp;
}
}
return root;
}