分析:
方法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;
}
};