117. Populating Next Right Pointers in Each Node II

最新推荐文章于 2020-07-11 10:02:10 发布

coffee-123

最新推荐文章于 2020-07-11 10:02:10 发布

阅读量217

点赞数

分类专栏：力扣经典题目文章标签：深度优先遍历

本文链接：https://blog.csdn.net/u014376961/article/details/53147605

版权

力扣同时被 2 个专栏收录

172 篇文章 0 订阅

订阅专栏

经典题目

78 篇文章 0 订阅

订阅专栏

这里写图片描述

可以用队列层序遍历，但空间复杂度是O（N）


class Solution {
public:
    void connect(TreeLinkNode *root) 
    {
        if(root==NULL)
            return;
        queue<TreeLinkNode *>q;
        q.push(root);
        q.push(NULL);
        while(!q.empty())
        {
            TreeLinkNode *temp=q.front();
            q.pop();
            if(temp)
            {
                temp->next=q.front();
                if(temp->left)
                    q.push(temp->left);
                if(temp->right)
                    q.push(temp->right);
            }
            else
            {
                if(q.empty())
                    return;
                q.push(NULL);
            }
        }
    }
};

第二种方式：深度优先遍历，但不同于116题的完美二叉树，这题要先把右子树递归，再递归左子树

class Solution 
{
public:
    void connect(TreeLinkNode *root) 
    {
        if(root==NULL)
            return;
        TreeLinkNode *curnext=root->next;
        TreeLinkNode *p=NULL;
        while(curnext)
        {
            if(curnext->left)
            {
                p=curnext->left;
                break;
            }
            if(curnext->right)
            {
                p=curnext->right;
                break;
            }
            curnext=curnext->next;
        }
        if(root->right)
            root->right->next=p;
        if(root->left)
        {
            if(root->right)
                root->left->next=root->right;
            else
                root->left->next=p;
        }
        connect(root->right);
        connect(root->left);
    }
};