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

最新推荐文章于 2024-09-14 22:25:23 发布

nuita

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量104

点赞数

文章标签：算法 c++ 数据结构 leetcode

本文链接：https://blog.csdn.net/qq_25155231/article/details/127461715

版权

二叉树队列广度优先搜索节点连接算法

关键词由CSDN通过智能技术生成

class Solution {
public:
    Node* connect(Node* root) {
        int size = 1,i = 0;
        Node* p = root;
        queue<Node*>s;
        if (p)s.push(p);
        while (!s.empty()) {
            p = s.front();
            s.pop();
            i++;
            if (p->left)s.push(p->left);
            if (p->right)s.push(p->right);
            if (i < size) {
                p->next = s.front();
            }
            else if (i == size)
                p->next = NULL, i = 0, size = s.size();
      
        }
        return root;
    }
};