【力扣刷题总结之117. 填充每个节点的下一个右侧节点指针 II】

相关标签

---

一、题目要求

二、题解和代码实现 

1.题解

非官方题解，非常详细的广度优先算法

2.代码实现

代码如下（示例）：

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

        LinkedList<Node> queue = new LinkedList<>();
        queue.add(root); //相当于把数据加入到队列尾部
        while (!queue.isEmpty()){

            int size = queue.size();//当前层的元素个数
            Node  nodePre;//初始化头节点
            Node node = null;//初始化最后一个节点
            for (int i = 0; i < size; i++) {
                if (i==0){//头节点
                    nodePre = queue.getFirst();//拿到当前队列首元素
                    queue.poll();//弹出首元素
                    node = nodePre;//设置最后结点为头结点
                }else {
                    nodePre= queue.getFirst();//拿到当前队列首元素
                    queue.poll();//弹出元素
                    node.next = nodePre;//设置node指针的next 指向弹出的元素
                    node = node.next;//移动node指针，一直指向到该层的最后一个元素
                }
                
                //左右子节点如果不为空就入队
                if (nodePre.left!=null){
                    queue.add(nodePre.left);
                }
                if (nodePre.right!=null){
                    queue.add(nodePre.right);
                }
            }

        }

        return root;
    }
}