【力扣算法题】填充节点右侧指针

【力扣算法题】填充节点右侧指针

---

---

题目介绍

给定一个 完美二叉树 ，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

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/