Populating Next Right Pointers in Each Node II 顺序链接二叉树每一层结点

最新推荐文章于 2021-01-29 16:34:55 发布

原创最新推荐文章于 2021-01-29 16:34:55 发布 · 353 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

146 篇文章

订阅专栏

二叉树的遍历

12 篇文章

订阅专栏

本文介绍了一种用于填充二叉树中每个节点的NEXT指针的算法,该指针指向其下一个右侧节点。通过递归方式实现,确保了左子节点的NEXT指针指向右子节点，并且在父节点的NEXT不为空时，继续链接到父节点右侧的子节点。

/**
* Definition for binary tree with next pointer.
* struct TreeLinkNode {
* int val;
* TreeLinkNode *left, *right, *next;
* TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
* };
*/
class Solution {
public:
void connect(TreeLinkNode *root) {
if(root==NULL) return;
if(root)
{
if(root->left)
{
root->left->next=root->right;
if(!root->right)
{
TreeLinkNode *node=root->next;
while(node&&root->left->next==NULL)
{
if(node->left)
root->left->next=node->left;
else if(!node->left)
root->left->next=node->right;
node=node->next;
}
}

}

if(root->right)
{
TreeLinkNode *node=root->next;
while(node&&root->right->next==NULL)
{
if(node->left)
root->right->next=node->left;
else if(!node->left)
root->right->next=node->right;
node=node->next;
}
}
connect(root->right);
connect(root->left);
}
}
};