力扣(LeetCode)116. 填充每个节点的下一个右侧节点指针(C++)

清墨韵染

于 2022-11-29 14:52:57 发布

阅读量374

点赞数 2

分类专栏：墨染leetcode 文章标签： leetcode c++ 算法模拟二叉树

本文链接：https://blog.csdn.net/Innocence02/article/details/128097939

版权

墨染leetcode 专栏收录该内容

138 篇文章 3 订阅

订阅专栏

模拟

这题可以直接操作根节点，我们保存根结点，用作最终返回值。

填充每个结点的 $n e x t$ 指针，其实是树的层序遍历。由于 $n e x t$ 指针的存在，我们可以做到 $O (1)$ 的空间复杂度。

算法:
从根结点出发，利用 $n e x t$ 指针同层右移，每个结点的左儿子的 $n e x t =$ 右儿子，右儿子的 $n e x t =$ 下一结点的左儿子。

p->left->next = p->right;//左儿子的 next=右儿子
p->right->next = p->next->left;//右儿子的 next=下一结点的左儿子

一层遍历结束后， $r o o t$ 走到左儿子，进入下一层遍历。

class Solution {
public:
    Node* connect(Node* root) {
        if(!root) return root;
        Node *temp = root;
        while(root->left){
            for(auto p = root;p;p = p ->next){
                p->left->next = p->right;
                if(p->next) p->right->next = p->next->left;
            }
            root = root->left;
        }
        return temp;
    }
};

时间复杂度 : $O (n)$ ， $n$ 是结点个数，每个结点至多被遍历一次，时间复杂度 $O (n)$ 。
空间复杂度 : $O (1)$ ，只使用常量级空间。