LeetCode | Populating Next Right Pointers in Each Node II（将每一层链接成一个链表）

最新推荐文章于 2023-05-28 09:45:57 发布

solar一抹阳光

最新推荐文章于 2023-05-28 09:45:57 发布

阅读量530

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/a45872055555/article/details/38570805

LeetCode 专栏收录该内容

150 篇文章

订阅专栏

本文介绍了一种在二叉树中填充每个节点的next指针指向其右侧相邻节点的方法。提出了两种解决方案，一种使用队列进行层级遍历，另一种利用树的结构特性，通过已连接的上一层节点来连接下一层节点。

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

Follow up for problem "Populating Next Right Pointers in Each Node".

What if the given tree could be any binary tree? Would your previous solution still work?

Note:

You may only use constant extra space.

For example,
Given the following binary tree,

After calling your function, the tree should look like:

         1 -> NULL
       /  \
      2 -> 3 -> NULL
     / \    \
    4-> 5 -> 7 -> NULL

题目解析：

本来这道题有个完全二叉树的，但感觉没必要，利用两种通用的方法都能解决，这里只写这道题了。

方案一：

利用程序遍历的思想，每一层之间的元素链接成链表，循环到队列为空为止。

class Solution {
public:
    void connect(TreeLinkNode *root) {
        if(root == NULL)
            return ;
        queue<TreeLinkNode *> qtree;
        qtree.push(root);
        while(!qtree.empty()){
            int n = qtree.size();
            TreeLinkNode *p = qtree.front();
            qtree.pop();    //记得要出队列，同时把子孩子入队列
            if(p->left) qtree.push(p->left);
            if(p->right) qtree.push(p->right);
            for(int i = 1;i < n;i++){
                TreeLinkNode *q = qtree.front();
                qtree.pop();
                if(q->left) qtree.push(q->left);
                if(q->right) qtree.push(q->right);
                p->next = q;
                p = q;
            }
            p->next = NULL;
        }
    }
};

方案二：

很巧妙的方法，其实也是一层一层的去链接，但是当处理第i层的时候，i-1层已经链接成一个链表，直接用cur = cur->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) {
        if(root == NULL) return ;
        root->next = NULL;
        TreeLinkNode* prevHead = root;
        while(true)
        {
            TreeLinkNode* cur = prevHead;
            TreeLinkNode* prev = NULL;
            while(cur != NULL)
            {
                if(cur->left != NULL)
                {
                    if(prev != NULL) prev->next = cur->left, prev = prev->next;
                    else prev = cur->left, prevHead = prev;
                }
                if(cur->right != NULL)
                {
                    if(prev != NULL) prev->next = cur->right, prev = prev->next;
                    else prev = cur->right, prevHead = prev;
                }
                cur = cur->next;
            }
            if(prev != NULL) prev->next = NULL;
            else break;
        }
    }
};