Leetcode 117. Populating Next Right Pointers in Each Node II（详细解析）

Leetcode 117. Populating Next Right Pointers in Each Node II

题目链接： Populating Next Right Pointers in Each Node II

难度：Medium

题目大意：

给出二叉树，给二叉树中的每个节点增加一个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;
        }
        Node cur=root;
        Node levelHead=new Node(0);//下一层的虚拟头结点，它的next指向这一层最左边的节点
        Node levelNode=levelHead;//用来遍历下一层的各个节点
        while(cur!=null){

            while(cur!=null){//遍历这一层
                if(cur.left!=null){
                    levelNode.next=cur.left;//注意！因为levelNode=levelHead,当levelNode.next指向cur.left的时候,levelHead.next也指向了cur.left。
                    levelNode=levelNode.next;
                }
                if(cur.right!=null){
                    levelNode.next=cur.right;
                    levelNode=cur.right;
                }
                cur=cur.next;
            }
            cur=levelHead.next;//进入下一层
            levelHead.next=null;//令其指向null，为指向cur的下一层做准备
            levelNode=levelHead;
        }
        return root;
    }
}