【LeetCode】117. 填充每个节点的下一个右侧节点指针 II（Java）

给定一个二叉树

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

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

初始状态下，所有 next 指针都被设置为 NULL。

进阶：

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

提示：

• 树中的节点数小于 6000
• -100 <= node.val <= 100

/*
// 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 null;

        //遍历每一层
        Node cur = root;
        while (cur != null) {
            //origin的next节点：cur下层的第一个节点
            Node origin = new Node(0);
            //pre：用于连接在cur下层连接的节点
            Node pre = origin;
            //遍历cur所在层，操作它的子节点
            while (cur != null) {
                if (cur.left != null) {
                    //pre连接上left节点
                    pre.next = cur.left;
                    //移到left节点
                    pre = pre.next;
                }
                if (cur.right != null) {
                    //pre连接上right节点
                    pre.next = cur.right;
                    pre = pre.next;
                }
                //cur指向下个节点，继续操作它的子节点
                cur = cur.next;
            }
            //跳出第二个循环，说明cur层已经走完，并且连接完它的子节点
            //cur指向下一层第一个节点
            cur = origin.next;
        }
        return root;
    }

}

思路来源题解区数据结构和算法大佬的题解，思路大体为把每一层当成一个链表，用一个指针记录第一个节点，用另一个指针把这一层的所有节点连接起来（cur为当前层，连接的是下一层的节点）。