116. 填充同一层的兄弟节点

最新推荐文章于 2020-10-15 16:29:56 发布

原创最新推荐文章于 2020-10-15 16:29:56 发布 · 123 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种在完美二叉树中填充next指针的方法，使得每个节点的next指针指向其右侧相邻的节点。通过递归地设置左孩子的next指向右孩子，右孩子的next指向父节点next的左孩子，实现了常数空间复杂度的要求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个二叉树

struct TreeLinkNode {
TreeLinkNode *left;
TreeLinkNode *right;
TreeLinkNode *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。

说明:

你只能使用额外常数空间。
使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。
你可以假设它是一个完美二叉树（即所有叶子节点都在同一层，每个父节点都有两个子节点）。
示例:

给定完美二叉树，

/
2 3
/ \ /
4 5 6 7
调用你的函数后，该完美二叉树变为：

 1 -> NULL

/
2 -> 3 -> NULL
/ \ /
4->5->6->7 -> NUL

解题：
将左孩子的next指向右孩子，右孩子的next指向其父节点的next（如果有的话）的左孩子，如果其父的next指向空了，就将该右孩子的next指向为空。
C++代码：
lass Solution {
public:
void connect(TreeLinkNode *root) {
if (!root) return;
if (root->left) root->left->next = root->right;
if (root->right) root->right->next = root->next ? root->next->left : NULL;
connect(root->left);
connect(root->right);

};