力扣LeetCode #117 填充每个节点的下一个右侧节点指针II（Connect）

- 题目描述

给定一个二叉树

struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。初始状态下，所有 next 指针都被设置为 NULL。

进阶：

• 你只能使用常量级额外空间。
• 使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

来源：LeetCode

- 示例

输入：root = [1,2,3,4,5,null,7]
输出：[1,#,2,3,#,4,5,7,#]

- 思路分析

和#116几乎一样，唯一的区别是#116里是完美二叉树，每一行节点个数都是上一行的两倍；本题中则不一定，因此需要多一个统计该层有多少个节点的步骤。

- JAVA实现

class Solution {
    public Node connect(Node root) {
        Node save = root;
        if(root == null) return root;
        Queue<Node> queue = new LinkedList();  
        queue.add(root);  
        recur(queue, 1);
        return root;
    }

    public void recur(Queue queue, int height) {
        int number = 0;  //记录下一行的节点个数
        int count = 0;
        while(count < height) {
            Node root = (Node)queue.poll();
            count++;
            if(count != height) root.next = (Node)queue.peek();  //说明不是最后一个
            else root.next = null;  //说明是这一行的最后一个
            if(root.left != null) {
                queue.add(root.left);
                number++;
            }
            if(root.right != null) {
                queue.add(root.right);
                number++;
            }  
        }
        if(queue.peek() == null) return;
        else recur(queue, number);  
    }
}