Populate Next Right Pointer in Each Node

来源:互联网 发布:网络平台贷款逾期 编辑:程序博客网 时间:2024/06/07 01:39

分析:

方法1: 如果是递归。可以考虑父节点为一个link list 来链接下面的子节点。如果父节点为空说明子节点也要链接完了。

先序遍历,也就是从上到下,从左到右依次链接。时间复杂度是O(n) 空间复杂度是 O(logn)

方法2: 如果是迭代。一般的思路是用一个queue来存放每一层的节点,在根据已有的层来安排下一层的节点。

要求是constant space 所以不能用queue。其实,这里题目有一个数据结构,list。而且节点也都存在,只需要相互之间链接就可以了。

同样是按照BFS,根据上一层的信息来安排下一层。关键是安排这一层list的head和指针。

时间复杂度是 O(n), 空间复杂度 O(1)


Code (C++):

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


0 0