leetcode116填充每个节点的下一个右侧节点指针

阿维的博客日记

于 2024-03-17 10:45:48 发布

阅读量257

点赞数 3

分类专栏： LeetCode题库文章标签：二叉树 BFS 广度优先搜索

本文链接：https://blog.csdn.net/weixin_46028606/article/details/136778366

版权

LeetCode题库专栏收录该内容

63 篇文章 0 订阅

订阅专栏

文章讲述了两种BFS（广度优先搜索）方法解决树结构连接问题，一种使用Queue直接更新节点指针，效率较高；另一种则是先存储所有子节点再连接，虽然详细但相对复杂。

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

思路：
BFS，层次遍历的时候，我用的Queue存储每层的所有元素，因此我先用Node* node保存Q的第一个元素，然后把Q的第一个pop掉，然后让node的next指向pop之后的Q的第一个元素，就顺理成章完成了本题！

BFS（题解的高效解法）

class Solution {
public:
    Node* connect(Node* root) {
        bfs(root);
        return root;
    }
    void bfs(Node* root){
        if(!root){
            return;
        }
        queue<Node*> qn;
        qn.push(root);
        while(!qn.empty()){
            int sz=qn.size();
            for(int i=0;i<sz;++i){
                Node* node=qn.front();
                qn.pop();
                if(node){
                    if(i<sz-1){
                        node->next=qn.front();
                    }
                    if(node->left){
                        qn.push(node->left);
                    }
                    if(node->right){
                        qn.push(node->right);
                    }
                }
            }

        }
    }
};

BFS（我的繁琐解法）

class Solution {
public:
    Node* connect(Node* root) {
        bfs(root);
        return root;
    }
    void bfs(Node* root){
        if(!root){
            return;
        }
        queue<Node*> qn;
        qn.push(root);
        while(!qn.empty()){
            int sz=qn.size();
            vector<Node*> v;
            for(int i=0;i<sz;i++){
                Node* node=qn.front();
                qn.pop();
                
                if(node){
                    if(node->left){
                        qn.push(node->left);
                        v.push_back(node->left);
                    }
                    if(node->right){
                        qn.push(node->right);
                        v.push_back(node->right);
                    }
                }

            }
            if(!v.empty()){
                for(int i=0;i<v.size()-1;++i){
                    v[i]->next=v[i+1];
                }
            }
        }
    }
};