LeetCode 117.Populating Next Right Pointers in Each Node II (填充同一层的兄弟节点 II)

最新推荐文章于 2021-12-25 19:04:31 发布

原创最新推荐文章于 2021-12-25 19:04:31 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #C #117 # Populating Next Right Pointer #填充同一层的兄弟节点 II

Leetcode 同时被 2 个专栏收录

144 篇文章

订阅专栏

Leetcode (C++解题报告)

142 篇文章

订阅专栏

本文介绍了一种在二叉树结构中填充next指针的方法，使得每个节点的next指针指向其右侧相邻的节点。通过一次BFS遍历实现，仅使用常数额外空间，满足题目要求。示例展示了从给定二叉树到填充后的结构变化。

题目描述：

给定一个二叉树

struct TreeLinkNode {
  TreeLinkNode *left;
  TreeLinkNode *right;
  TreeLinkNode *next;
}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。

说明:

你只能使用额外常数空间。
使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

示例:

给定二叉树，

调用你的函数后，该二叉树变为：

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

AC C++ Solution:
这是BFS遍历。现在指针是当前层的遍历，head是下一层的最左边元素，tail是下一层的最右边元素。将指针在当前层移动并填充其子层。（当前层遍历完毕移动到下一层的头部，这种关系已经在上一层遍历中填充填充）

class Solution {
public:
    void connect(TreeLinkNode *root) {
        TreeLinkNode *now, *tail, *head;
        
        now = root;
        head = tail = NULL;
        while(now) {
            if(now->left) {
                if(tail)    
                    tail = tail->next = now->left;
                else
                    head = tail = now->left;
            }
            if(now->right) {
                if(tail)
                    tail = tail->next = now->right;
                else
                    head = tail = now->right;
            }
            
            if(!(now = now->next)) {    //若now没有next,则将now设置为下一行头节点,开始遍历下一行节点
                now = head;
                head = tail = NULL;
            }
        }
    }
};