【Leetcode】NO.116.填充每个节点的下一个右侧节点指针 (C++&Python)[二叉树]

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

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

思路

在层序遍历的基础上每一成的节点出来之前将next指针指向队列头部，i<size-1，最后一个节点是null,而且初始化所有节点的next都会null;

代码

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node* _left, Node* _right, Node* _next)
        : val(_val), left(_left), right(_right), next(_next) {}
};
*/

class Solution {
public:
    Node* connect(Node* root) {
        // 层序遍历，初始状态都是null

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

使用递归方法
先处理一层节点的情况，之后使用递归方法实现
Python 版本

"""
# Definition for a Node.
class Node:
    def __init__(self, val: int = 0, left: 'Node' = None, right: 'Node' = None, next: 'Node' = None):
        self.val = val
        self.left = left
        self.right = right
        self.next = next
"""

class Solution:
    def connect(self, root: 'Optional[Node]') -> 'Optional[Node]':
        if root and root.left:
            root.left.next = root.right
            if root.next:
                root.right.next = root.next.left
            self.connect(root.left)
            self.connect(root.right)

        return root