二叉树的题目,我认为二叉树必须要掌握递归的三种遍历算法,以及层序遍历算法才能做二叉树题目。这道题目我的想法是:
因为在二叉树每一层中,next指针指向的是的当前节点的右边的节点,所以,使用层序遍历,在每个节点出队后,指向当前层的下一节点,这样,遍历完整个树,next指针也就设置好了。
看看代码如何实现:
/*
// 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 root;
Queue<Node> queue = new LinkedList<Node>();
queue.offer(root);
int num = 1;//用于遍历每层的节点数
int k = 0;//用于计算每层节点数量
while(true){
k = 0; //每层开始时置0
for(int i=0; i < num; i++){ //遍历每层节点
Node node = queue.poll();//出队一个节点
if(queue.peek() != null && i != num-1) //如果队列顶还有元素,且不是下一层元素将next指向该节点
{
node.next = queue.peek();
}else //否则指向空
{
node.next = null;
}
if(node.left != null)//加入下层节点
{
queue.offer(node.left);
k++;//下层节点数量加1
}
if(node.right != null)//加入下层节点
{
queue.offer(node.right);
k++;//下层节点数量加1
}
}
if(k == 0)break;//如果下层没有节点,退出循环
num = k;//赋值给控制遍历的变量
}
return root;
}
}