LeetCode: Populating Next Right Pointers in Each Node II

AIvin24

于 2014-08-30 10:15:30 发布

阅读量621

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/AIvin24/article/details/38942119

LeetCode 专栏收录该内容

144 篇文章

订阅专栏

本文详细介绍了如何利用特定博客的思路，通过迭代方式实现一棵树中相邻节点的连接，同时保持空间复杂度为O(1)，并附上对应的代码实现。

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

思路：这个题目参考了这篇博客的思路，在每一层中，先用一个指针 P 记录需要连接的节点，碰到下一个节点 Q 后，令 P->next = Q，而且更新P = Q，注意这是P不为空的情况，在每一层中，可以依次遍历每个节点，当一个节点的左孩子或者右孩子非空时，这这个孩子节点是下一层的 P的初始值。这样就完成了空间复杂度为O(1)的实现。

code:

class Solution {
public:
    void connect(TreeLinkNode *root) {
        while(root){
            TreeLinkNode * nextLayer = NULL;
            TreeLinkNode * needToLink = NULL;
            TreeLinkNode * p = root;
            while(p && nextLayer == NULL){
                    nextLayer = p->left != NULL ? p->left : p->right;
                    p = p->next;
            }
            needToLink = nextLayer;
            while(root){
                if(root->left != NULL){
                    if(needToLink != root->left) needToLink->next = root->left;
                    needToLink = root->left;
                }
                if(root->right != NULL){
                    if(needToLink != root->right) needToLink->next = root->right;
                    needToLink = root->right;
                }
                root = root->next;
            }
            root = nextLayer;
        }
    }
};