【力扣算法题】填充节点右侧指针
题目介绍
给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
题解
1. 广度优先(个人解法/力扣官方解法1)
首先这题的说明是有歧义的,注意题目中“如果找不到下一个右侧节点,则将 next 指针设置为 NULL”当时给我的想法就是,假设在同一个深度上,我们有[1, NULL, 3, 4]这样四个节点,那么最后连接完应该是这样"1-> NULL, 3 -> 4 ->NULL",然而实际上,它题目的意思是让你直接跳过中间的NULL,也就是说结果应该是这样的“1 -> 3 -> 4 -> NULL”,直接给我坑到了
这个歧义解决了就简单了,十分明显的广度优先题目,对每一层深度的节点都分别访问,连接就行
C++源码:
Node* connect(Node* root) {
queue<Node*> q;
if(root == nullptr){ return nullptr; }
q.push(root);
int counter;
while(!q.empty())
{
counter = q.size();
while(counter > 0)
{
Node* curNode = q.front();
q.pop();
if(counter > 1)
{
curNode->next = q.front();
}
if(curNode->left != nullptr)
{
q.push(curNode->left);
}
if(curNode->right != nullptr)
{
q.push(curNode->right);
}
counter--;
}
}
return root;
}
2. 利用next逐层遍历
二叉树的特点就是每一个节点都有2个子节点,那么左节点的next必然是右节点,于是我们可以利用这个性质,对每个节点的子节点进行连接。
这时候会出现一个问题,不是一个父节点的子节点之间要如何连接。我们可以通过next来获取当前父节点的下一个子节点,从而连接
C++源码:
Node* connect(Node* root) {
if (root == nullptr) {
return root;
}
// 从根节点开始
Node* leftmost = root;
while (leftmost->left != nullptr) {
// 遍历这一层节点组织成的链表,为下一层的节点更新 next 指针
Node* head = leftmost;
while (head != nullptr) {
// CONNECTION 1
head->left->next = head->right;
// CONNECTION 2
if (head->next != nullptr) {
head->right->next = head->next->left;
}
// 指针向后移动
head = head->next;
}
// 去下一层的最左的节点
leftmost = leftmost->left;
}
return root;
}
// 作者:LeetCode-Solution
// 链接:https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node/solution/tian-chong-mei-ge-jie-dian-de-xia-yi-ge-you-ce-2-4/