LeetCode刷题日志-117填充每个节点的下一个右侧指针II

二叉树的题目，我认为二叉树必须要掌握递归的三种遍历算法，以及层序遍历算法才能做二叉树题目。这道题目我的想法是：
因为在二叉树每一层中，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;
    }   
}