二叉树转换为单链表-Python

问题：

# 给你二叉树的根结点 root ，请你将它展开为一个单链表： 
# 
#  
#  展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。 
#  展开后的单链表应该与二叉树 先序遍历 顺序相同。 
#  
# 
#  
# 
#  示例 1： 
# 
#  
# 输入：root = [1,2,5,3,4,null,6]
# 输出：[1,null,2,null,3,null,4,null,5,null,6]
#  
# 
#  示例 2： 
# 
#  
# 输入：root = []
# 输出：[]
#  
# 
#  示例 3： 
# 
#  
# 输入：root = [0]
# 输出：[0]
#  
# 
#  
# 
#  提示： 
# 
#  
#  树中结点数在范围 [0, 2000] 内 
#  -100 <= Node.val <= 100 
#  
# 
#  
# 
#  进阶：你可以使用原地算法（O(1) 额外空间）展开这棵树吗？ 
#  Related Topics 栈 树 深度优先搜索 链表 二叉树 
#  👍 907 👎 0


# leetcode submit region begin(Prohibit modification and deletion)
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right

解决：迭代。

思路：

# 我们先观察例子，可以发现，左子树展开成链表连接在根节点，而右子树展开链表是紧跟在左子树展开的链表后面。这里使用递归的方法来解决。

# 使用后序遍历（递归）的具体做法：
# 先找到左子树的最右节点；
# 当找到左子树的最右节点时，令其指向根的右子树；
# 此时，再让根的右子树，指向根节点的左子树；
# 最后，令根节点的左子树为 None，循环直至右子树为空。

def flatten(root):
    if not root:
        return
    flatten(root.left)
    flatten(root.right)
    # 如果存在左子树
    if root.left != None:
        p = root.left    # 找到左孩子节点
        while p.right != None:    # 如果有右孩子,一直找到最右边节点
            p = p.right
        p.right = root.right    # p节点的右孩子指向根节点的右孩子，右子串直接拿过去
        root.right = root.left  # 根节点的右孩子指向根节点的左孩子
        root.left = None        # 根节点左指针为赋值为空