填充每个节点的下一个右侧节点指针 II
题目描述:
给定一个二叉树
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
输入:{"$id":"1","left":{"$id":"2","left":{"$id":"3","left":null,"next":null,"right":null,"val":4},"next":null,"right":{"$id":"4","left":null,"next":null,"right":null,"val":5},"val":2},"next":null,"right":{"$id":"5","left":null,"next":null,"right":{"$id":"6","left":null,"next":null,"right":null,"val":7},"val":3},"val":1}
输出:{"$id":"1","left":{"$id":"2","left":{"$id":"3","left":null,"next":{"$id":"4","left":null,"next":{"$id":"5","left":null,"next":null,"right":null,"val":7},"right":null,"val":5},"right":null,"val":4},"next":{"$id":"6","left":null,"next":null,"right":{"$ref":"5"},"val":3},"right":{"$ref":"4"},"val":2},"next":null,"right":{"$ref":"6"},"val":1}
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。
提示:
你只能使用常量级额外空间。
使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
解题思路:采用递归方法,这一题是116的升级版,那一题的规定是完全二叉树,所以对于左右节点操作比较简单,但是这一题是一颗标准的二叉树,所以对于左右节点的操作,要仔细去分析,还有要注意的就是如果需要父节点的next,一定要走到上一层的第一个有孩子的结点,不能只是父节点的next*(与116的不同之处),还有就是递归先从右节点开始(因为如果先从左节点开始的话,父节点的next并不能完整的设置完,会有缺失,相反,从右节点开始,这样从右往左递归,next正好能完全设置完。)
class Solution {
public:
Node* connect(Node* root) {
if(!root)return root;
root->next=NULL;
create_point(root,root->left,root->right);
return root;
}
void create_next(Node* p,Node* node){
while(p){
if(p->left){
node->next=p->left;
break;
}
if(p->right){
node->next=p->right;
break;
}
p=p->next;
}
if(!p)node->next=NULL;
}
void create_point(Node* prior,Node* l,Node* r){
if(!l&&!r)return;
Node* p=prior->next;
if(l){
if(r)l->next=r;
else if(p)create_next(p,l);
else l->next=NULL;
}
p=prior->next;
if(r){
if(p)create_next(p,r);
else r->next=NULL;
}
if(r)create_point(r,r->left,r->right);
if(l)create_point(l,l->left,l->right);
}
};