力扣 116. 填充每个节点的下一个右侧节点指针、力扣 117. 填充每个节点的下一个右侧节点指针 II

幸运征服者

已于 2023-07-28 00:38:36 修改

阅读量84

点赞数

分类专栏： LeetCode Algorithms 文章标签：算法 leetcode 数据结构 c++

于 2023-07-27 21:28:24 首次发布

本文链接：https://blog.csdn.net/m0_57307125/article/details/131968908

版权

LeetCode Algorithms 专栏收录该内容

58 篇文章 0 订阅

订阅专栏

该文介绍了一种通过层序遍历来处理二叉树节点的算法，将每个节点的next指针指向其右侧的节点。方法包括创建队列，将根节点入队，然后在循环中处理每一层的节点，更新next指针，并将未处理的子节点入队。时间复杂度和空间复杂度均为O(n)。

摘要由CSDN通过智能技术生成

这道题要求将每一个节点的next指针指向右侧的节点。可以使用层序遍历，一层一层的处理节点。

时间复杂度：O(n)，空间复杂度：O(n)

思路：

创建一个队列，如果不为空树，则将根节点（第一个节点）入队。
构建一个循环，循环终止条件为队列中没有需要处理的节点，即将树中全部节点处理完。
再构建一个嵌套循环，用于处理当前层级的所有节点，循环终止条件是处理完当前级别的所有节点。
1. 在此循环中，如果当前处理节点是该层最后一个节点，则next指针指向NULL。否则，指向队列中的下一个front节点，即该节点的右侧节点。
2. 如果当前处理节点有左子节点，则将其放入队列中，在下一个大循环里作为下一层级的节点处理。右子节点同理。
大循环结束，返回根节点root。

代码：

/*
// 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) {
        queue<Node*> que;
        if (root != NULL)
            que.push(root);
        while (!que.empty()) {
            int size = que.size();
            while (size--) {
                Node* node = que.front();
                que.pop();
                if (size == 0)
                    node->next = NULL;
                else {
                    node->next = que.front();
                }
                if (node->left)
                    que.push(node->left);
                if (node->right)
                    que.push(node->right);
            }
        }
        return root;
    }
};