[leetcode]python3 算法攻略-填充同一层的兄弟节点

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

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

说明:

• 你只能使用额外常数空间。
• 使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。
• 你可以假设它是一个完美二叉树（即所有叶子节点都在同一层，每个父节点都有两个子节点）。

方案一：利用层次遍历，连接每一层的节点

# Definition for binary tree with next pointer.
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None

class Solution:
    # @param root, a tree link node
    # @return nothing
    def connect(self, root):
        if not root:
            return
        stack = [root]
        while stack:
            l = len(stack)
            for i in range(l):
                cur = stack.pop(0)
                if i < l - 1:
                    cur.next = stack[0]
                if cur.left:
                    stack.append(cur.left)
                if cur.right:
                    stack.append(cur.right)
        return

方案二：递归

class Solution:
    # @param root, a tree link node
    # @return nothing
    def connect(self, root):   
        if root and root.left:  #如果该节点有左右子树
            root.left.next = root.right #先将左右子树的节点用next指针链接起来（由于是完美二叉树，所以如果有左子树，肯定有右子树）
            if root.next:   #如果该节点的next指针指向的节点不为None，则指向next指向的节点的左子树的节点
                root.right.next = root.next.left
            else:
                root.right.next = None
            self.connect(root.left)
            self.connect(root.right)