填充每个节点的下一个右侧节点指针（BFS简单实现）_bfs 填充二叉树每个节点得下个右侧节点指针-CSDN博客

本文链接：https://blog.csdn.net/weixin_43857345/article/details/109156699

本文介绍了一种算法，用于填充完美二叉树中每个节点的next指针，使其指向同一层的下一个节点。通过广度优先搜索实现，先获取每层节点数再进行操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。初始状态下，所有 next 指针都被设置为 NULL。

示例：
在这里插入图片描述

注意点

1、获取每一层节点数必须在所有操作之前。

实现

/*
// 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) {
        return bfs(root);
    }

    public Node bfs(Node root) {
        if (null == root) {
            return root;
        }

		// 队列用于存放每一层的节点
        Queue<Node> queue = new LinkedList<>();
        // 初始化根节点
        queue.offer(root);

		// 存在节点
        while (!queue.isEmpty()) {
        	// 获取每一层的节点数
            int queueSize = queue.size();
            // 获取前节点
            Node pre = queue.poll();
            // 判断该节点是否存在子节点，存在则进队（可以只判断是否存在左节点，因为是完全二叉树）
            if (null != pre.left) {
                queue.offer(pre.left);
            }
            if (null != pre.right) {
                queue.offer(pre.right);
            }
            // 存在下一节点
            for (int i = 1; i < queueSize; i ++) {
            	// 获取当前节点
                Node cur = queue.poll();
                // 前节点指向当前节点
                pre.next = cur;
                // 更新前节点
                pre = cur;
                // 判断该节点是否存在子节点，存在则进队
                if (null != cur.left) {
                    queue.offer(cur.left);
                }
                if (null != cur.right) {
                    queue.offer(cur.right);
                }
            }
            // 每一层最后节点指向null（可以省略，因为默认就指向null）
            pre.next = null;
        }

        return root;
    }
}