Populating Next Right Pointers in Each Node

最新推荐文章于 2022-02-28 18:16:11 发布

原创最新推荐文章于 2022-02-28 18:16:11 发布 · 336 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

111 篇文章

订阅专栏

本文探讨了两种填充二叉树节点的算法实现，一种采用广度优先搜索，并通过队列辅助；另一种则利用递归特性，巧妙地将同一父节点下的左右子节点连接起来。后者更节省空间且优雅。

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

不难的一道题，但是我始终没想出如何用常数空间来完成……

我使用了广搜的办法，思路特别简单，要点是每层放入队列以后在最后面插入一个NULL节点来作标记位。

/**
 * 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) {
        if (root == NULL)
            return;
       deque<TreeLinkNode *> d;
       d.push_back(root);
       d.push_back(NULL);
       while (!d.empty()) {
           TreeLinkNode *t = d.front();
           d.pop_front();
           if (t->left != NULL && t->right != NULL) {
                d.push_back(t->left);
                d.push_back(t->right);
           }
           if (d.front() == NULL) {
               d.pop_front();
               if (d.empty())
                    break;
               d.push_back(NULL);
               continue;
           }
           else {
               t->next = d.front();
           }
       }
    }
};

网上还看到一个更好的方法，把需要连接的节点在他们共同祖先那一层一起处理，利用递归来结题。

http://blog.youkuaiyun.com/havenoidea/article/details/12840497

/**
 * 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) {
        if(!root)return;
        if(root->left)
        {
            TreeLinkNode *l=root->left,*r=root->right;
            while(l!=NULL)
            {
                l->next=r;
                l=l->right;
                r=r->left;
            }
            connect(root->left);
            connect(root->right);
        }
    }
};
// blog.youkuaiyun.com/havenoidea

http://oj.leetcode.com/problems/populating-next-right-pointers-in-each-node/