前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Populating Next Right Pointers in Each Node

Populating Next Right Pointers in Each Node

作者头像
Tyan
发布2019-05-25 22:38:26
6680
发布2019-05-25 22:38:26
举报
文章被收录于专栏:SnailTyan

1. Description

Populating Next Right Pointers in Each Node
Populating Next Right Pointers in Each Node

2. Solution

  • Version 1
代码语言:javascript
复制
/**
 * 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*> q1;
        queue<TreeLinkNode*> q2;
        q1.push(root);
        TreeLinkNode* pre = nullptr;
        TreeLinkNode* current = nullptr;
        while(!q1.empty()) {
            current = q1.front();
            q1.pop();
            if(current->left && current->right) {
                q2.push(current->left);
                q2.push(current->right);
            }
            if(pre) {
                pre->next = current;
            }
            pre = current;
            if(q1.empty()) {
                pre->next = nullptr;
                pre = nullptr;
                q1 = q2;
                q2 = {};
            }
        }
    }
};
  • Version 2
代码语言:javascript
复制
/**
 * 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*> q;
        q.push(root);
        connect(q);
    }

private:
    void connect(queue<TreeLinkNode*>& q) {
        TreeLinkNode* pre = nullptr;
        TreeLinkNode* current = nullptr;
        queue<TreeLinkNode*> q2;
        while(!q.empty()) {
            current = q.front();
            q.pop();
            if(current->left && current->right) {
                q2.push(current->left);
                q2.push(current->right);
            }
            if(pre) {
                pre->next = current;
            }
            pre = current;
        }
        pre->next = nullptr;
        if(!q2.empty()) {
            connect(q2);
        }
    }
};
  • Version 3
代码语言:javascript
复制
/**
 * 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;
        }
        while(root) {
            TreeLinkNode* current = root;
            while(current && current->left) {
                current->left->next = current->right;
                if(current->next) {
                    current->right->next = current->next->left;
                }
                current = current->next;
            }
            root = root->left;
        }
    }
};

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018年12月05日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. Description
  • 2. Solution
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档