LeetCode 116 填充每个节点的下一个右侧节点指针 Python3解法

原题来自于:链接
给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

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

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

示例：

解法一

如果不考虑空间复杂度的话，当然是选择层次遍历啦！开辟一个数组保存每一层的节点，然后用next指针将其链接即可。不多说，直接看代码吧！

def connect(root):
    # 一种不考虑空间复杂的做法
    # 开辟空间 保存层次遍历的节点 然后再链接起来
    if not root：
        return None
    queue = [root]
    depthNodes = 2
    while True:
        tmp = []
        if len(tmp) != depthNodes:
            # 利用满二叉树的性质
            cur = queue.pop(0)
            if not cur.left:
                # 意味着结束
                return root
            tmp.append(cur.left)
            tmp.append(cur.right)
        for i in range(1, depthNodes):
            tmp[i-1].next = tmp[i]
        depthNodes *= 2
        queue = tmp[:]

有一个depthNodes作为每一层结束的标志信息。如果保存节点的数组长度达到了depthNodes，则进入下一层，由于是完美二叉树，depthNodes是以2为底的指数函数。

解法二

我们当然不能够仅仅满足现状，利用常数空间复杂度的解法当然存在。本质上我们就需要不断地将需要的节点链接即可。可以考虑以下代码块，假设当然节点为cur：

if cur.left:
	cur.left.next = cur.right

这段代码所做的就是将自己的两个儿子(left、right)给连起来。
再看这段代码：

if cur.right and cur.next:
	cur.right.next = cur.next.left

很明显，这段代码所做的就是将兄弟节点(二叉树的中间)给连上。

有了这个主要的逻辑处理，我们就不难写出递归和迭代的代码了：

# 递归写法
def connect(root):
    # 一种所谓的递归方法
    # 先处理单个节点的逻辑 然后递归调用
    if not root:
        return
    if root.left:
        root.left.next = root.right
        if root.right and root.next:
            # root.right可以省略 因为满二叉树的定义
            root.right.next = root.next.left
    connect(root.left)
    connect(root.right)
    return root

# 迭代写法
def connect(root):
    # 一种所谓的迭代方法
    # 处理逻辑同上 只是需要一个指针pre保存每一层节点的起始位置 便于往下迭代
    if not root:
        return None
    pre = root
    while pre:
        cur = pre
        while cur:
            # 迭代直到每一层所有的节点都已经遍历
            if cur.left:
                cur.left.next = cur.right
            if cur.right and cur.next:
                cur.right.next = cur.next.left
            cur = cur.next
        # 下一层
        pre = pre.left
    return root

解法三

有意思的是这个“拉拉链”解法，参考的Leetcode题解中的一种。
我们把完美二叉树看成一件大衣好了，现在需要做的就是将所有的口都给拉上。比如中间这个拉链：

left = root.left
right = root.right
while left:
	left.next = right
	# 去下一层
	left = left.right
	right = right.left

事实上，个人觉得这叫“扣扣子”解法更为妥当，因为它实质上是从上往下进行连接，符合大多数人的扣扣子逻辑(拉拉链不是从下往上么o(￣︶￣)o？？？)。
好了，递归地进行左右子树扣扣子拉拉链解法：

def connect(root):
    if not root:
        return
    left = root.left
    right = root.right
    while left:
        # 存在没拉上拉链的节点
        left.next = right
        # 更新节点到下一层
        left = left.right
        right = right.left
    self.connect(root.left)
    self.connect(root.right)
    return root

乱花渐欲迷人眼，浅草才能没马蹄