填充每个节点的下一个右侧节点指针 II（Leetcode-117）- BFS

最新推荐文章于 2024-02-25 07:30:00 发布

LLWeirdy、

最新推荐文章于 2024-02-25 07:30:00 发布

阅读量66

点赞数

分类专栏：刷题笔记文章标签： dfs 二叉树队列 leetcode

本文链接：https://blog.csdn.net/weixin_44855409/article/details/118882808

版权

刷题笔记专栏收录该内容

45 篇文章 0 订阅

订阅专栏

题目

原题连接：https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node-ii/

知识点

广度优先搜索
队列

思路

这道题和二叉树的层序遍历 II（Leetcode-107）有异曲同工之处，明白了那道题，这道就方便多了。简单来说，while循环一次遍历一层节点，for 循环逐个元素同一层元素逐个访问。
因此，只需要用last记录”前驱“（这里的前驱是同一层的前面一个节点！），last->next 指向“后继”即可（Line 33）
PS：在指向后继时，需要判断 if (last != nullptr) ，因为每一层的第一个节点 last = NULL

代码

class Solution {
public:
    queue<Node *> q;

    Node *connect(Node *root) {
        if (root == nullptr) return root;
        q.push(root);
        Node *u = nullptr;
        while (!q.empty()) {
            int cnt = q.size();
            Node *last = nullptr;
            for (int i = 0; i < cnt; i++) {
                u = q.front();
                q.pop();
                if (last != nullptr)  last->next = u;
                if (u->left != nullptr) q.push(u->left);
                if (u->right != nullptr) q.push(u->right);
                last = u;
            }
            last->next = nullptr;  // 每一层的最后一个节点指向空
        }
        return root;
    }
};