leetcode_116 填充每个节点的下一个右侧节点指针I

public Node connect(Node root) {
    Deque<Node> q = new LinkedList();
    if(root == null) return root;
    q.offer(root);
    while(!q.isEmpty()) {
        int size = q.size();
        for(int i = 0; i < size; i++) {
            Node node1 = q.poll();
            Node node2 = q.peek();
            if(node1.left != null) q.offer(node1.left);
            if(node1.right != null) q.offer(node1.right);
            if(i != size-1) node1.next = node2;
            else node1.next = null;
        }
    }
    return root;
}