Leetcode: Populating Next Right Pointers in Each Node

logic_nut

于 2013-10-16 05:50:29 发布

阅读量614

点赞数

分类专栏：面试题

本文链接：https://blog.csdn.net/logic_nut/article/details/12769295

版权

面试题专栏收录该内容

50 篇文章

订阅专栏

本文介绍了一种使用队列实现的算法，用于解决二叉树节点链接问题，具体步骤包括初始化队列、遍历节点并连接相邻节点。

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

http://oj.leetcode.com/problems/populating-next-right-pointers-in-each-node/

/**
 * 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) {}
 * };
 */
 using namespace std;
class Solution {
public:
    void connect(TreeLinkNode *root) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if(root==NULL) return;
        queue<TreeLinkNode*> nodes;
        nodes.push(root);
        nodes.push(NULL);
        while(nodes.size()!=0){
            TreeLinkNode* current=nodes.front();
            nodes.pop();
            if(current!=NULL){
                current->next=nodes.front();
                if(current->left!=NULL) nodes.push(current->left);
                if(current->right!=NULL)nodes.push(current->right);
            }
            else{
                if(nodes.size()!=0)
                    nodes.push(NULL);
            }
        }
    }
};