填充每个节点的下一个右侧节点指针

填充每个节点的下一个右侧节点指针

描述

给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

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

思路和代码

递归，对于一个结点root，

• 若root的左子树不为空，则完美二叉树一定有右子树，将root->left->next = root->right
• 若root的next不为空，说明root有右兄弟，也就是root的右孩子要指向叔叔的左孩子
• 递归左孩子，递归右孩子

class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL)
            return root;
        if(root->left)
        {
            root->left->next = root->right;
            if(root->next)
            {
                root->right->next = root->next->left;
            }
        }
        connect(root->left);
        connect(root->right);
        return root;
    }
};

也可以用层次遍历去做，和下面一题代码一样

if(root == NULL)
            return root;
        queue<Node*> q;
        q.push(root);
        while(!q.empty())
        {
            Node* pre = NULL;
            int size = q.size();
            for(int i=0; i<size; i++)
            {
                Node* cur = q.front();
                q.pop();
                if(i > 0)
                    pre->next = cur;
                pre = cur;
                if(cur->left)
                    q.push(cur->left);
                if(cur->right)
                    q.push(cur->right);
            }
        }
        return root;

填充每个节点的下一个右侧节点指针 II

描述

不同于上一题，上一题是完全二叉树，这一题是任意的二叉树
给定一个二叉树

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

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

思路和代码

层次遍历的思路，层次遍历正好是一层一个接一个。

class Solution {
public:
    Node* connect(Node* root) {
        //层次遍历
        if(root == NULL)
            return root;
        queue<Node*> q;
        q.push(root);
        while(!q.empty())
        {
            Node* pre = NULL;
            int size = q.size();
            for(int i=0; i<size; i++)
            {
                Node* cur = q.front();
                q.pop();
                if(i > 0)
                    pre->next = cur;
                pre = cur;
                if(cur->left)
                    q.push(cur->left);
                if(cur->right)
                    q.push(cur->right);
            }
        }
        return root;
    }
};