Populating Next Right Pointers in Each Node II

xuxumoi

于 2015-07-12 19:29:12 发布

阅读量213

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： Tree Depth-First Search LeetCode Queue

本文链接：https://blog.youkuaiyun.com/xsloop/article/details/46852573

LeetCode 专栏收录该内容

107 篇文章

订阅专栏

本文介绍了一种使用队列实现的算法，该算法能够有效地将二叉树中同一层的相邻节点通过next指针连接起来。通过遍历二叉树的每一层，并利用队列保存每一层的节点，实现了节点之间的相互连接。

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

/**
 * 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 == NULL)
            return;
        queue<TreeLinkNode*> q;
        q.push(root);
        q.push(NULL);
        
        while(q.size() > 1)
        {
            TreeLinkNode* t = q.front();
            q.pop();
            
            if(t == NULL)
            {
                q.push(NULL);
                continue;
            }
            
            t->next = q.front();
            
            if(t->left)
                q.push(t->left);
            if(t->right)
                q.push(t->right);
            
        }
    }
};