题解 | #填充每个节点指向最右节点的next指针#

填充每个节点指向最右节点的next指针

https://www.nowcoder.com/practice/fdbd05d647084fcf9be78444e231998b

/**
 * Definition for binary tree with next pointer.
 * struct TreeLinkNode {
 *  int val;
 *  TreeLinkNode *left, *right, *next;
 *  TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
 * };
 */
#include <cstddef>
#include <queue>
class Solution {
  public:
    void connect(TreeLinkNode* root) {
        if (root == nullptr)
            return ;
        queue<TreeLinkNode*> qu;
        queue<TreeLinkNode*> jl;
        qu.push(root);
        bool hou = false;
        while (qu.size() > 0 || jl.size() > 0) {
            if (hou == false) {
                TreeLinkNode* t;
                t = qu.front();
                qu.pop();
                if (qu.size() == 0) {
                    t->next = nullptr;
                    hou = true;
                } else
                    t->next = qu.front();
                if (t->left != nullptr) {

                    jl.push(t->left);
                }
                if (t->right != nullptr) {

                    jl.push(t->right);
                }
            } else {
                TreeLinkNode* t;
                t = jl.front();
                jl.pop();
                if (jl.size() == 0) {
                    t->next = nullptr;
                    hou = false;
                } else
                    t->next = jl.front();
                if (t->left != nullptr) {

                    qu.push(t->left);
                }
                if (t->right != nullptr) {

                    qu.push(t->right);
                }
            }

        }
    }
};

全部评论

相关推荐

不要停下啊:大二打开牛客,你有机会开卷了,卷起来,去找课程学习,在牛客上看看大家面试笔试都需要会什么,岗位有什么需求就去学什么,努力的人就一定会有收获,这句话从来都经得起考验,像我现在大三了啥也不会,被迫强行考研,炼狱难度开局,啥也不会,找工作没希望了,考研有丝丝机会
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-03 18:22
投了几百份简历,专业和方向完全对口,都已读不回。尝试改了一下学校,果然有奇效。
steelhead:这不是很正常嘛,BOSS好的是即便是你学院本可能都会和聊几句,牛客上学院本机会很少了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务