117.填充每个节点的下一个右侧节点 II

十年一觉尘与土2023-11-03 8:54

题目来源：

leetcode题目，网址：117. 填充每个节点的下一个右侧节点指针 II - 力扣（LeetCode）

解题思路：

按层遍历时修改 next 指针即可，每一层除最后一个元素的next 指针指向空外，皆指向同层下一个元素。

解题代码：

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node* _left, Node* _right, Node* _next)
        : val(_val), left(_left), right(_right), next(_next) {}
};
*/

class Solution {
public:
    Node* connect(Node* root) {
        if(root==nullptr){
            return nullptr;
        }
        queue<Node*> q1;
        q1.push(root);
        while(!q1.empty()){
            int size=q1.size();
            for(int i=0;i<size;i++){
                Node* temp=q1.front();
                q1.pop();
                if(i!=size-1){
                    temp->next=q1.front();
                }
                if(temp->left!=nullptr){
                    q1.push(temp->left);
                }
                if(temp->right!=nullptr){
                    q1.push(temp->right);
                }
            }
        }
        return root;
    }
};

复制代码

总结：

官方题解给出了两种解法。第一种是层次遍历。第二种是使用已建立的 next 指针，若在遍历第 i 层的节点时，为第 i+1 层建立 next 指针，那么在遍历下一层时，只需知道下一层的首个节点即可通过已建立的 next 指针遍历整层。