⭐算法入门⭐《广度优先搜索》中等02 —— LeetCode 116. 填充每个节点的下一个右侧节点指针

本文链接：https://blog.csdn.net/WhereIsHeroFrom/article/details/118657057

本文介绍了一篇关于C语言学习和算法刷题的博客内容，强调了使用广度优先搜索解决二叉树问题，提供了一个填充完美二叉树节点next指针的解题报告，包括思路、时间复杂度和代码详解。博主还组织了一个刷题群，鼓励读者通过刷题提升算法能力，特别是针对‘LeetCode’的题目进行练习。

摘要由CSDN通过智能技术生成

🙉饭不食，水不饮，题必须刷🙉
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
LeetCode 太难？先看简单题！ 🧡《C语言入门100例》🧡
数据结构难？不存在的！ 🌳《数据结构入门》🌳
LeetCode 太简单？算法学起来！ 🌌《夜深人静写算法》🌌

一、题目

1、题目描述

给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。初始状态下，所有 next 指针都被设置为 NULL。
样例输入：

样例输出：

2、基础框架

c++ 版本给出的基础框架代码如下：

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node* _left, Node* _right, Node* _next)
        : val(_val), left(_left), right(_right), next(_next) {}
};
*/

class Solution {
public:
    Node* connect(Node* root) {
        
    }
};

其中Node代表完美二叉树的树节点，除了数据域val，指针域left代表左子树根节点指针，right代表右子树根节点指针，next默认为NULL，是需要我们通过算法来进行填充的。

3、原题链接

LeetCode 116. 填充每个节点的下一个右侧节点指针

二、解题报告

1、思路分析

对于二叉树类型的问题，99% 都是考递归（也就是深度优先搜索），然而，这个题很有意思。这是一棵完美二叉树，所以我们看到题目要求做的事情是将树的每一层用链表链起来，对于层序遍历，其实是广搜。所以就是打着深搜的幌子进行广搜了。
首先，从树根节点开始进行层序遍历（广度优先搜索）；
同时，对于进入队列中的元素进行编号， $1$ 号元素单独成为一个链； $\to 3$ 号成为一条链； $\to 7$ 号成为一条链； $2^n \to 2^{n+1} - 1$ 成为一条链；

2、时间复杂度

每个节点入队一次，出队一次，总的时间复杂度为 $O (n)$ 。

3、代码详解

class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL) {
            return NULL;               // (1)
        }
        Node *last = NULL;             // (2)
        queue<Node*> que;
        que.push(root);                // (3)
        int cnt = 0;
        while(!que.empty()) {
            Node *p = que.front();     // (4)
            que.pop();
            ++cnt;

            if(last) {                 // (5)
                last->next = p;
            }

            if( (cnt + 1) & cnt ) {    // (6)
                last = p;
            }else {
                last = NULL;
            }

            if(p->left) {              // (7)
                que.push(p->left);
                que.push(p->right);
            }
        }
        return root;                   // (8)
    }
};

$(1)$ 注意判断空树的情况；
$(2)$ last指针用来记录每一条链表的上一个节点；
$(3)$ 将根节点塞入队列，进行广搜；
$(4)$ 取出队列中的一个节点作为当前节点，并且计数器加一；
$(5)$ 如果不是当前层的第一个节点，则让链表上一个节点指向当前节点；
$(6)$ 这一步判定就比较微妙了。我们可以这么考虑，对于一个二进制表有 $k$ 个 1 的数，即：
$(\underbrace{11...11}_{\rm k})_2$
我们给它加上1，得到：
$(1\underbrace{00...00}_{\rm k})_2$
两个数进行位与，就会得到 0，而除了二进制表示全是 1 的情况，其它不会出现位与等于 0 的情况。由此表明，对于一个数 $x$ ：二进制表示全是一是 ((x + 1) & x ) == 0的充要条件。
结合之前的图：
当计数器为 1、3、7、15、… 、 $2^n-1$ 时，last指针要置为 NULL代表一条链的结束，下一条链的开始；
$(7)$ 当不是叶子节点时，将左右子树塞入队列；
$(8)$ 返回根节点；

三、本题小知识

1）((x + 1) & x ) == 0表示 $x = 2^n-1$ ；
2）二叉树的题不一定要用深度优先搜索来解决，也可以是广度优先搜索；

四、加群须知

相信看我文章的大多数都是「大学生」，能上大学的都是「精英」，那么我们自然要「精益求精」，如果你还是「大一」，那么太好了，你拥有大把时间，当然你可以选择「刷剧」，然而，「学好算法」，三年后的你自然「不能同日而语」。
那么这里，我整理了「几十个基础算法」 的分类，点击开启：

🌌《算法入门指引》🌌
如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

大致题集一览：

在这里插入图片描述

为了让这件事情变得有趣，以及「照顾初学者」，目前题目只开放最简单的算法 「枚举系列」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有一半成员刷完 「枚举系列」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「夜深人静写算法」专家团 的一员。
不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「主要经济来源」，在你成为神的路上，「不会索取任何」。
🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
入门级C语言真题汇总 🧡《C语言入门100例》🧡
几张动图学会一种数据结构 🌳《画解数据结构》🌳
组团学习，抱团生长 🌌《算法入门指引》🌌
竞赛选手金典图文教程 💜《夜深人静写算法》💜