题目:
给定一个完美二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
解题思路:
方法1: 递归
二叉树的递归解法,需要从该节点的角度进行实现,不要考虑堆栈;
代码按照前序遍历模板进行,有两个场景需要考虑:
- 该节点的左节点 需要指向右节点;
- 该节点的右节点(如果是非根节点),需要指向该节点的左节点;
- 如果该节点是叶节点,不进行操作。
struct Node *connect(struct Node *root)
{
// 特殊场景判断
if (!root) {
return root;
}
// 考虑根节点需要做的事情
//如果左节点不存在,说明该节点为叶节点,不进行操作;只从父亲节点进行操作
if (!root->left) {
return root;
}
// 场景1:左节点 指向右节点;
root->left->next = root->right;
// 场景2: 右节点,指向 兄弟父节点的左节点(难点)
if (root->next) {
root->right->next = root->next->left;
}
// 树的先序遍历
connect(root->left);
connect(root->right);
return root;
}