LeetCode 116. Populating Next Right Pointers in Each Node

最新推荐文章于 2025-08-16 00:47:01 发布

yurio7r

最新推荐文章于 2025-08-16 00:47:01 发布

阅读量419

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/MrSiz/article/details/53284942

LeetCode 专栏收录该内容

178 篇文章

订阅专栏

本文介绍了一种使用队列实现树的层次遍历的方法，通过将每层节点依次加入队列并进行遍历，实现了节点间的连接。该算法适用于二叉树结构，并通过递归的方式确保了每个节点都被正确地连接到同一层级的下一个节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述

给出一棵树，把每一层的节点连接起来

解决

利用queue存每一层的节点，然后遍历即可。

/**
 * 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) {}
 * };
 */
class Solution {
public:
    void connect(TreeLinkNode *root) {
        if (!root)
            return ;
        queue<TreeLinkNode *> que;
        que.push(root);
        root -> next = NULL;
        int cnt = 1;
        while (!que.empty())
        {
            TreeLinkNode *zz = NULL;
            int t = 0;
            for (int i = 0; i < cnt; ++i)
            {
                TreeLinkNode *tmp = que.front();
                //cout << tmp -> val << endl;
                que.pop();
                if (tmp -> left)
                {
                    que.push(tmp -> left);
                    ++t;
                }
                if (tmp -> right)
                {
                    que.push(tmp -> right);
                    ++t;
                }
                if (i == 0)
                {
                    zz= tmp;
                    continue;
                }
                else if (i != cnt - 1)
                {
                    zz -> next = tmp;
                    zz = tmp;
                }
                else if (i == cnt - 1)
                {
                    zz -> next = tmp;
                    tmp -> next = NULL;
                }
            }
            cnt = t;
        }
        return ;
    }
};