leetcode 116 --- 填充每个节点指向最右节点的next指针

最新推荐文章于 2024-07-23 23:23:58 发布

原创最新推荐文章于 2024-07-23 23:23:58 发布 · 248 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#队列 #二叉树 #数据结构 #算法

刷题专栏收录该内容

29 篇文章

订阅专栏

本文介绍了一种在完美二叉树中填充每个节点的next指针的方法，使得同一层相邻节点能够通过next指针相连。文章提供了一种不使用额外队列的解决方案，仅利用常数级额外空间实现遍历。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 题目

填充每个节点指向最右节点的next指针,

填充所有节点的next指针，指向最接近它的同一层右边节点。如果没有同一层没有右边的节点，则应该将next指针设置为NULL。

初始时，所有的next指针都为NULL

注意：

你只能使用常量级的额外内存空间
可以假设给出的二叉树是一个完美的二叉树(即，所有叶子节点都位于同一层，而且每个父节点都有两个孩子节点)。

2 解法

因为此题为按层来填充每一个节点的next为其同层的右节点(如果存在的话)

2.1 用队列的方法

2.2 不用队列

因为需要将每一层用next指针链接好, 所以遍历的思想是,链接好的一层遍历的时候把其下一层链接好, 在每一层遍历时,因为要链接其下一层的各个节点,所以要用一个last指针将上一个被链接为next的节点标记好,且要标记好下一层的头节点,以便于上一层遍历完遍历下一层,所以有:

/**
 * Definition for binary tree with next pointer.
 * struct TreeLinkNode {
 *  int val;
 *  TreeLinkNode *left, *right, *next;
 *  TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
 * };
 */
class Solution {
public:
    void connect(TreeLinkNode *root) {
        TreeLinkNode *start(root);
        while (start) {
            TreeLinkNode *nextStart(NULL), *last(NULL);
            for (auto p = start; p != NULL; p = p->next) {
                if (p->left) {
                    if (!nextStart)
                        nextStart = p->left;
                    if (last)
                        last->next = p->left;
                    last = p->left;
                }
                if (p->right) {
                    if (!nextStart)
                        nextStart = p->right;
                    if (last)
                        last->next = p->right;
                    last = p->right;
                }
            }
            start = nextStart;
        }
    }
};