题目
给定一个二叉树
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
进阶:
你只能使用常量级额外空间。
使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
示例:
输入:root = [1,2,3,4,5,null,7]
输出:[1,#,2,3,#,4,5,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化输出按层序遍历顺序(由 next 指针连接),’#’ 表示每层的末尾。
- 方法一:迭代
/*
// Definition for a Node.
class Node {
public:
int val;
Node* left;
Node* right;
Node* next;
Node() : val(0), left(NULL), right(NULL), next(NULL) {}
Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}
Node(int _val, Node* _left, Node* _right, Node* _next)
: val(_val), left(_left), right(_right), next(_next) {}
};
*/
class Solution {
public:
Node* connect(Node* root) {
if(root==NULL) return root;
Node *cur = root;
//设置每一层的next指针,cur指向上一层,pre指向当前层
while(cur!=NULL){
Node dummy(0);//dummy为该层链表的头结点
Node *pre = &dummy;//pre最初指向dummy
while(cur!=NULL){//遍历上一层,设置该层的next指针
if(cur->left!=NULL){//如果cur有左孩子
pre->next = cur->left;
pre = pre->next;
}
if(cur->right!=NULL){//如果cur有右孩子
pre->next = cur->right;
pre = pre->next;
}
cur = cur->next;//cur指向它的后继
}
cur = dummy.next;//该层遍历完,为下一层做准备
}
return root;
}
};
-
时间复杂度O(n)
-
空间复杂度O(1)
-
思路
- 第一层不需要设置next指针,以后的每一层借助上一层的next指针设置该层的next指针。
- cur指向根节点,第一个while循环中遍历每一层,一次循环设置一层的next节点。
- 设置该层的next指针前,先设置一个头节点,作为连接后链表首节点的前驱,在设置完该层后利用头结点设置遍历下一次时的起始节点。
- 设置该层指针时,pre始终指向设置next指针后的节点链表的最后一个节点。如果cur有左孩子,将pre的后继设置为cur的左孩子,pre指向它的后继;如果cur有右孩子,pre的后继设置为cur的右孩子。cur指向cur的后继,直到cur为空,表示该层已经设置完毕。
-
方法二:方法一的递归版
/*
// Definition for a Node.
class Node {
public:
int val;
Node* left;
Node* right;
Node* next;
Node() : val(0), left(NULL), right(NULL), next(NULL) {}
Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}
Node(int _val, Node* _left, Node* _right, Node* _next)
: val(_val), left(_left), right(_right), next(_next) {}
};
*/
class Solution {
public:
Node* connect(Node* root) {
if(root==NULL) return root;
Node *cur = root;
Node dummy(0);
Node *pre = &dummy;
//设置该层next指针
while(cur!=NULL){
if(cur->left!=NULL){
pre->next = cur->left;
pre = pre->next;
}
if(cur->right!=NULL){
pre->next = cur->right;
pre = pre->next;
}
cur = cur->next;
}
connect(dummy.next);//递归调用设置下一层的next指针
return root;
}
};
- 时间复杂度O(n)
- 空间复杂度O(logn)
- 思路
- 同方法一,只是把最后一层while循环改为了递归调用